diff --git a/image_ref/main.py b/image_ref/main.py
index a28e3f70ab7d8192816e784f198213a3c94daf32..c335df083565efa96dcbc70bb99a1bd7d3a23d73 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -115,8 +115,8 @@ def run_duo(args):
     plt.savefig('../output/training_plot_contrastive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model))
     #load and evaluate best model
     load_model(model, args.save_path)
-    make_prediction_duo(model,data_test_batch, '../output/confusion_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model),
-                        '../output/confidence_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model))
+    make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model),
+                        'output/confidence_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model))
 
 
 def make_prediction_duo(model, data, f_name, f_name2):