diff --git a/test/external/external_test_datasets.py b/test/external/external_test_datasets.py
index a07e4e7d92..5e8bc1533f 100644
--- a/test/external/external_test_datasets.py
+++ b/test/external/external_test_datasets.py
@@ -157,7 +157,7 @@ class TestOpenImagesDataset(ExternalTestDatasets):
     ref_dataloader = self._create_ref_dataloader(base_dir, ann_file, "val")
     transform = GeneralizedRCNNTransform(img_size, img_mean, img_std)
 
-    for ((tinygrad_img, _), (ref_img, _)) in zip(tinygrad_dataloader, ref_dataloader):
+    for ((tinygrad_img, _, _), (ref_img, _)) in zip(tinygrad_dataloader, ref_dataloader):
       ref_img, _ = transform(ref_img.unsqueeze(0))
       np.testing.assert_equal(tinygrad_img.numpy(), ref_img.tensors.transpose(1, 3).numpy())