diff --git a/vall_e/train.py b/vall_e/train.py
index 972ae45..260a5e7 100755
--- a/vall_e/train.py
+++ b/vall_e/train.py
@@ -251,7 +251,7 @@ def run_eval(engines, eval_name, dl, args=None):
 	}
 
 	try:
-		for engine in engines:	
+		for name, engine in engines.items():
 			if engine.wandb is not None:
 				engine.wandb.log({
 					f'{eval_name}.loss.mstft': stats['loss'],