diff --git a/answer_classifier_cached_features/select_best_model.py b/answer_classifier_cached_features/select_best_model.py
index 9d0d9dea6e6c4073c785188af1d2ebb5ff817d31..893fc898184488251c61bfa62bb911455dd9f0fc 100644
--- a/answer_classifier_cached_features/select_best_model.py
+++ b/answer_classifier_cached_features/select_best_model.py
@@ -45,6 +45,8 @@ def create_batch_generator():
         constants.vqa_train_held_out_qids,
         constants.vocab_json,
         constants.vqa_answer_vocab_json,
+        constants.object_labels_json,
+        constants.attribute_labels_json,
         constants.image_size,
         constants.num_region_proposals,
         constants.num_negative_answers,