diff --git a/Python/DeepSSMUtilsPackage/DeepSSMUtils/__init__.py b/Python/DeepSSMUtilsPackage/DeepSSMUtils/__init__.py
index 25b070feeb..a56fb1c9c9 100644
--- a/Python/DeepSSMUtilsPackage/DeepSSMUtils/__init__.py
+++ b/Python/DeepSSMUtilsPackage/DeepSSMUtils/__init__.py
@@ -50,7 +50,7 @@ def analyzeResults(out_dir, DT_dir, prediction_dir, mean_prefix):
 	return avg_distance
 
 def get_image_registration_transform(fixed_image_file, moving_image_file, transform_type='rigid'):
-	itk_transform = image_utils.get_image_registration_transform(fixed_image_file, moving_image_file, transform_type='rigid')
+	itk_transform = image_utils.get_image_registration_transform(fixed_image_file, moving_image_file, transform_type)
 	return itk_transform
 
 def testPytorch():