diff --git a/answer_classifier_cached_features/select_best_model.py b/answer_classifier_cached_features/select_best_model.py
index 385c21f3cb8a5fa91b0372932d27528c0944a9fc..038b5ab630c3d1f98e027abbafa0fd930e9601c7 100644
--- a/answer_classifier_cached_features/select_best_model.py
+++ b/answer_classifier_cached_features/select_best_model.py
@@ -237,13 +237,14 @@ def eval_model(model_to_eval, results_json):
         constants.answer_batch_size,
         constants.image_size,
         constants.num_negative_answers,
-        constants.answer_embedding_dim,
         constants.answer_regularization_coeff,
         constants.answer_batch_size*constants.num_region_proposals,
         0,
         0,
         0,
         constants.answer_obj_atr_loss_wt,
+        constants.answer_ans_loss_wt,
+        constants.answer_mil_loss_wt,
         resnet_feat_dim=constants.resnet_feat_dim,
         training=False)