diff --git a/object_attribute_classifier_cached_features/select_best_model.py b/object_attribute_classifier_cached_features/select_best_model.py
index 198ac93ed468b0413315875de210ea068d77f2ea..f250ea7f58989017f9e324c9e7d35c2e46214b37 100644
--- a/object_attribute_classifier_cached_features/select_best_model.py
+++ b/object_attribute_classifier_cached_features/select_best_model.py
@@ -48,7 +48,7 @@ def create_batch_generator():
         constants.object_labels_json,
         constants.attribute_labels_json,
         constants.regions_json,
-        constants.genome_train_held_out_region_ids,
+        constants.genome_val_region_ids,
         constants.image_size,
         channels=3,
         resnet_feat_dim = constants.resnet_feat_dim,