diff --git a/combo/main.py b/combo/main.py
index 0bc6126b0787ba9cf362c5022b41100c18159576..c0365f36e45e55d8b0cad9e94cac7e432b674ac4 100644
--- a/combo/main.py
+++ b/combo/main.py
@@ -176,7 +176,7 @@ def _get_predictor() -> predictors.Predictor:
     )
 
     return predictors.Predictor.from_archive(
-        archive, FLAGS.predictor_name, extra_args= {"lambo_model_name" : FLAGS.lambo_model_name}
+        archive, FLAGS.predictor_name, "validation", True, {"lambo_model_name" : FLAGS.lambo_model_name}
     )