diff --git a/image_ref/main.py b/image_ref/main.py
index 27565f206975154bc9fa37b560f0ae8cf64691dd..738ea392d06c6400a2394e5c6d5f47201f740277 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -232,10 +232,10 @@ def run_duo(args):
     plt.ylim(0, 1.05)
     plt.show()
 
-    plt.savefig('../output/training_plot_contrastive_noise_{}_lr_{}_model_{}_{}.png'.format(args.noise_threshold,args.lr,args.model,args.model_type))
+    plt.savefig('output/training_plot_contrastive_noise_{}_lr_{}_model_{}_{}.png'.format(args.noise_threshold,args.lr,args.model,args.model_type))
     #load and evaluate best model
     load_model(model, args.save_path)
-    make_prediction_duo(model,data_test, '../output/confusion_matrix_contractive_noise_{}_lr_{}_model_{}_{}.png'.format(args.noise_threshold,args.lr,args.model,args.model_type))
+    make_prediction_duo(model,data_test, 'output/confusion_matrix_contractive_noise_{}_lr_{}_model_{}_{}.png'.format(args.noise_threshold,args.lr,args.model,args.model_type))
 
 
 def make_prediction_duo(model, data, f_name):