diff --git a/pvnet_app/model_configs/pydantic_models.py b/pvnet_app/model_configs/pydantic_models.py
index e5e0a08..8f37bf1 100644
--- a/pvnet_app/model_configs/pydantic_models.py
+++ b/pvnet_app/model_configs/pydantic_models.py
@@ -100,12 +100,10 @@ def get_all_models(
 
     models = config_pvnet_v2_model(models)
 
-    print(len(models.models))
     if get_ecmwf_only:
         log.info("Using ECMWF model only")
         models.models = [model for model in models.models if model.ecmwf_only]
 
-    print(len(models.models))
     if get_day_ahead_only:
         log.info("Using Day Ahead model only")
         models.models = [model for model in models.models if model.day_ahead]
@@ -113,12 +111,10 @@ def get_all_models(
         log.info("Not using Day Ahead model")
         models.models = [model for model in models.models if not model.day_ahead]
 
-    print(len(models.models))
     if not run_extra_models and not get_day_ahead_only and not get_ecmwf_only:
         log.info("Not running extra models")
         models.models = [model for model in models.models if model.name == "pvnet_v2"]
 
-    print(len(models.models))
     if use_ocf_data_sampler:
         log.info("Using OCF Data Sampler")
         models.models = [model for model in models.models if model.uses_ocf_data_sampler]
diff --git a/tests/test_app.py b/tests/test_app.py
index 9b789b2..3c9345f 100644
--- a/tests/test_app.py
+++ b/tests/test_app.py
@@ -175,6 +175,7 @@ def test_app_no_sat(
         os.environ["RUN_EXTRA_MODELS"] = "True"
         os.environ["SAVE_GSP_SUM"] = "True"
         os.environ["DAY_AHEAD_MODEL"] = "False"
+        os.environ["USE_OCF_DATA_SAMPLER"] = "True"
 
         # Run prediction
         # Thes import needs to come after the environ vars have been set