diff --git a/register.py b/register.py
index dc9c39e..ea89d97 100644
--- a/register.py
+++ b/register.py
@@ -91,7 +91,7 @@ def __init__(self, dim):
 
     def add_initial_transform(self, transform, param_scaling=None):
         if param_scaling is None:
-            param_scaling = np.ones((transforms.get_param_count(),))
+            param_scaling = np.ones((transform.get_param_count(),))
         self.initial_transforms.append(transform)
         self.transforms_param_scaling.append(param_scaling)