diff --git a/image_ref/main.py b/image_ref/main.py
index cbb8986ed53e15f1e4890afba3905573114472b7..a06cac935b56c34d4f25fa1bad81ccd1a951490b 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -169,11 +169,11 @@ def run_duo(args):
         plt.tight_layout()
 
         plt.show()
-        plt.savefig('output/training_plot_contrastive_{}.png'.format(args.positive_prop))
+        plt.savefig(args.base_out+'_training_plot.png')
 
     # load and evaluate best model
     load_model(model, args.save_path)
-    if args.args.dataset_test_dir is not None :
+    if args.dataset_test_dir is not None :
         make_prediction_duo(model, data_test_batch,args.base_out+'_confusion_matrix_test.png',
                             args.base_out+'confidence_matrix_.png')