diff --git a/Phase 3/utils.py b/Phase 3/utils.py
index 0891b3d..b94056e 100644
--- a/Phase 3/utils.py	
+++ b/Phase 3/utils.py	
@@ -277,7 +277,7 @@ def calculate_label_representatives(fd_collection, label, feature_model):
             img_fds[feature_model]
         ).flatten()  # get the specific feature model's feature vector
         for img_fds in fd_collection.find(
-            {"true_label": label, "$mod": [2,0]}
+            {"true_label": label, "image_id": {"$mod": [2,0]}}
         )  # repeat for all images
     ]