diff --git a/image_ref/main.py b/image_ref/main.py
index 3dda6c88a54560e971cfdb686b24b463907f8474..704f625d7440e6cb8ff6b563e5495d1c87e6de0d 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -137,8 +137,8 @@ def run_duo(args):
 
     #load and evaluate best model
     load_model(model, args.save_path)
-    make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.prop),
-                        'output/confidence_matrix_contractive_{}.png'.format(args.prop))
+    make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.positive_prop),
+                        'output/confidence_matrix_contractive_{}.png'.format(args.positive_prop))
 
 
 def make_prediction_duo(model, data, f_name, f_name2):