diff --git a/image_ref/sweep_train.py b/image_ref/sweep_train.py
index fd544fe21261c453228e25b18936746347697b83..d2a273ae07cbd75bde1624e00d89303a709c751b 100644
--- a/image_ref/sweep_train.py
+++ b/image_ref/sweep_train.py
@@ -130,7 +130,7 @@ def run_duo(args):
             val_loss.append(loss)
             val_acc.append(acc)
             val_cont_acc.append(acc_contrastive)
-        wdb.finish()
+    wdb.finish()
 
 
 if __name__ == '__main__':