diff --git a/projects/4-fashion-autoencoder/train.py b/projects/4-fashion-autoencoder/train.py
index 46223aae3..3f2d95b10 100644
--- a/projects/4-fashion-autoencoder/train.py
+++ b/projects/4-fashion-autoencoder/train.py
@@ -42,8 +42,8 @@ def log_images(epoch, logs):
     pred_data = np.clip(model.predict(test_data), 0, 1)
     wandb.log({
             "examples": [
-                wandb.Image(np.hstack([data, pred_data[i]]), caption=str(i))
-                for i, data in enumerate(test_data)]
+                wandb.Image(np.hstack([data, pred]), caption=str(i))
+                for i, (pred, data) in zip(test_data, pred_data)]
         }, commit=False)