diff --git a/sam3/train/data/sam3_image_dataset.py b/sam3/train/data/sam3_image_dataset.py
index c5a1c83..c956d7b 100644
--- a/sam3/train/data/sam3_image_dataset.py
+++ b/sam3/train/data/sam3_image_dataset.py
@@ -328,7 +328,7 @@ class CustomCocoDetectionAPI(VisionDataset):
                 f"Number of queries in stage {stage} is {num_queries}, expected {num_queries_per_stage}"
             )
 
-        for query_id, query in enumerate(queries):
+        for query in queries:
             h, w = id2imsize[query["image_id"]]
             if (
                 "input_box" in query