diff --git a/spec/schemas/ml._common.yaml b/spec/schemas/ml._common.yaml
index 39c8cb97e..2eb2536c2 100644
--- a/spec/schemas/ml._common.yaml
+++ b/spec/schemas/ml._common.yaml
@@ -114,8 +114,13 @@ components:
           type: string
           description: The model state.
           enum:
-            - DEPLOY_FAILED
             - REGISTERING
+            - REGISTERED
+            - DEPLOYING
+            - DEPLOYED
+            - PARTIALLY_DEPLOYED
+            - UNDEPLOYED
+            - DEPLOY_FAILED
         planning_worker_node_count:
           type: number
           description: The planning worker node count.