diff --git a/apps/worker/worker.py b/apps/worker/worker.py
index 00a435e..bfaa535 100644
--- a/apps/worker/worker.py
+++ b/apps/worker/worker.py
@@ -171,9 +171,9 @@ def parse_and_validate_payload(payload: dict) -> tuple[dict, List[str]]:
     
     # Validate model
     if "model" in payload:
-        valid_models = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost", "CatBoost_V2"]
-        if payload["model"] not in valid_models:
-            errors.append(f"Invalid model: {payload['model']}. Must be one of {valid_models}")
+        from contracts import VALID_MODELS
+        if payload["model"] not in VALID_MODELS:
+            errors.append(f"Invalid model: {payload['model']}. Must be one of {VALID_MODELS}")
     
     # Validate kernel
     if "smoothing_kernel" in payload: