diff --git a/image_ref/main.py b/image_ref/main.py
index 160ff67ca2fdc5ab12e379f8e302af21b70cea5f..82cd816a4192f96e5f3daf5314ee1df043bb2042 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -103,7 +103,7 @@ def run_duo(args):
             val_acc.append(acc)
             if loss < best_loss :
                 save_model(model,args.save_path)
-                loss = acc
+                best_loss = loss
     # plot and save training figs
     plt.clf()
     plt.subplot(2, 1, 1)