diff --git a/train_decoder.py b/train_decoder.py
index 057ca6d..76b49c2 100644
--- a/train_decoder.py
+++ b/train_decoder.py
@@ -331,7 +331,7 @@ def train(
             sample = 0
             average_loss = 0
             timer = Timer()
-            for i, (img, emb, txt) in enumerate(dataloaders["val"]):
+            for i, (img, emb, *_) in enumerate(dataloaders["val"]):
                 sample += img.shape[0]
                 img, emb = send_to_device((img, emb))