diff --git a/image_ref/main.py b/image_ref/main.py
index 8e33af8755807021cf06ffafc35f99e19f7ef32e..017daeca73499f4ec29b7d2329b17c393f0ee41f 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -135,12 +135,12 @@ def run_duo(args):
 
 
     plt.show()
-    plt.savefig('../output/training_plot_contrastive_{}.png'.format(args.positive_prop))
+    plt.savefig('output/training_plot_contrastive_{}.png'.format(args.positive_prop))
 
     #load and evaluate best model
     load_model(model, args.save_path)
-    make_prediction_duo(model,data_test_batch, '../output/confusion_matrix_contractive_{}_bis.png'.format(args.positive_prop),
-                        '../output/confidence_matrix_contractive_{}_bis.png'.format(args.positive_prop))
+    make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}_bis.png'.format(args.positive_prop),
+                        'output/confidence_matrix_contractive_{}_bis.png'.format(args.positive_prop))
 
 
 def make_prediction_duo(model, data, f_name, f_name2):