diff --git a/image_ref/main.py b/image_ref/main.py
index 236da5f79096f7b33b8a6ed611d49ed514675a70..eaf030dca3abe6018aab659d44f9d22aa9413b1b 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -184,7 +184,7 @@ def run_duo(args):
     load_model(model, args.save_path)
     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_test.png')
+                            args.base_out+'_confidence_matrix_test.png')
 
     make_prediction_duo(model, data_val_batch,args.base_out+'_confusion_matrix_val.png',
                         args.base_out+'_confidence_matrix_val.png')