diff --git a/output/schema/schema.json b/output/schema/schema.json
index bbf6f7717c..47fc6f06c5 100644
--- a/output/schema/schema.json
+++ b/output/schema/schema.json
@@ -152944,29 +152944,7 @@
         "name": "DeploymentAllocationState",
         "namespace": "ml._types"
       },
-      "specLocation": "ml/_types/TrainedModel.ts#L288-L301"
-    },
-    {
-      "kind": "enum",
-      "members": [
-        {
-          "name": "starting"
-        },
-        {
-          "name": "started"
-        },
-        {
-          "name": "stopping"
-        },
-        {
-          "name": "failed"
-        }
-      ],
-      "name": {
-        "name": "DeploymentAssignmentState",
-        "namespace": "ml._types"
-      },
-      "specLocation": "ml/_types/TrainedModel.ts#L303-L308"
+      "specLocation": "ml/_types/TrainedModel.ts#L273-L286"
     },
     {
       "kind": "enum",
@@ -152982,13 +152960,17 @@
         {
           "description": "The deployment is preparing to stop and deallocate the model from the relevant nodes.",
           "name": "stopping"
+        },
+        {
+          "description": "The deployment is on a failed state and must be re-deployed.",
+          "name": "failed"
         }
       ],
       "name": {
-        "name": "DeploymentState",
+        "name": "DeploymentAssignmentState",
         "namespace": "ml._types"
       },
-      "specLocation": "ml/_types/TrainedModel.ts#L273-L286"
+      "specLocation": "ml/_types/TrainedModel.ts#L288-L305"
     },
     {
       "kind": "interface",
@@ -157037,7 +157019,7 @@
         "name": "RoutingState",
         "namespace": "ml._types"
       },
-      "specLocation": "ml/_types/TrainedModel.ts#L350-L371"
+      "specLocation": "ml/_types/TrainedModel.ts#L347-L368"
     },
     {
       "kind": "enum",
@@ -157860,7 +157842,7 @@
           }
         }
       ],
-      "specLocation": "ml/_types/TrainedModel.ts#L402-L417"
+      "specLocation": "ml/_types/TrainedModel.ts#L399-L414"
     },
     {
       "kind": "interface",
@@ -157918,7 +157900,7 @@
           }
         }
       ],
-      "specLocation": "ml/_types/TrainedModel.ts#L373-L391"
+      "specLocation": "ml/_types/TrainedModel.ts#L370-L388"
     },
     {
       "kind": "interface",
@@ -158029,7 +158011,7 @@
           }
         }
       ],
-      "specLocation": "ml/_types/TrainedModel.ts#L315-L348"
+      "specLocation": "ml/_types/TrainedModel.ts#L312-L345"
     },
     {
       "kind": "interface",
@@ -158425,7 +158407,7 @@
           }
         }
       ],
-      "specLocation": "ml/_types/TrainedModel.ts#L393-L400"
+      "specLocation": "ml/_types/TrainedModel.ts#L390-L397"
     },
     {
       "kind": "interface",
@@ -158771,7 +158753,7 @@
           "type": {
             "kind": "instance_of",
             "type": {
-              "name": "DeploymentState",
+              "name": "DeploymentAssignmentState",
               "namespace": "ml._types"
             }
           }
@@ -159046,7 +159028,7 @@
           }
         }
       ],
-      "specLocation": "ml/_types/TrainedModel.ts#L419-L421"
+      "specLocation": "ml/_types/TrainedModel.ts#L416-L418"
     },
     {
       "kind": "interface",
@@ -159067,7 +159049,7 @@
           }
         }
       ],
-      "specLocation": "ml/_types/TrainedModel.ts#L423-L425"
+      "specLocation": "ml/_types/TrainedModel.ts#L420-L422"
     },
     {
       "kind": "interface",
@@ -159101,7 +159083,7 @@
           }
         }
       ],
-      "specLocation": "ml/_types/TrainedModel.ts#L427-L436"
+      "specLocation": "ml/_types/TrainedModel.ts#L424-L433"
     },
     {
       "kind": "interface",
@@ -159264,7 +159246,7 @@
         "name": "TrainingPriority",
         "namespace": "ml._types"
       },
-      "specLocation": "ml/_types/TrainedModel.ts#L310-L313"
+      "specLocation": "ml/_types/TrainedModel.ts#L307-L310"
     },
     {
       "kind": "interface",
diff --git a/output/typescript/types.ts b/output/typescript/types.ts
index 15a0e13646..e9e3ef2ad2 100644
--- a/output/typescript/types.ts
+++ b/output/typescript/types.ts
@@ -14054,9 +14054,7 @@ export interface MlDelayedDataCheckConfig {
 
 export type MlDeploymentAllocationState = 'started' | 'starting' | 'fully_allocated'
 
-export type MlDeploymentAssignmentState = 'starting' | 'started' | 'stopping' | 'failed'
-
-export type MlDeploymentState = 'started' | 'starting' | 'stopping'
+export type MlDeploymentAssignmentState = 'started' | 'starting' | 'stopping' | 'failed'
 
 export interface MlDetectionRule {
   actions?: MlRuleAction[]
@@ -14646,7 +14644,7 @@ export interface MlTrainedModelDeploymentStats {
   rejected_execution_count: integer
   reason: string
   start_time: EpochTime<UnitMillis>
-  state: MlDeploymentState
+  state: MlDeploymentAssignmentState
   threads_per_allocation: integer
   timeout_count: integer
 }
diff --git a/specification/ml/_types/TrainedModel.ts b/specification/ml/_types/TrainedModel.ts
index e781e4a166..439444e403 100644
--- a/specification/ml/_types/TrainedModel.ts
+++ b/specification/ml/_types/TrainedModel.ts
@@ -93,7 +93,7 @@ export class TrainedModelDeploymentStats {
   /** The epoch timestamp when the deployment started. */
   start_time: EpochTime<UnitMillis>
   /** The overall state of the deployment. */
-  state: DeploymentState
+  state: DeploymentAssignmentState
   /** The number of threads used be each allocation during inference. */
   threads_per_allocation: integer
   /** The sum of `timeout_count` for all nodes in the deployment. */
@@ -270,40 +270,37 @@ export enum TrainedModelType {
   pytorch
 }
 
-export enum DeploymentState {
+export enum DeploymentAllocationState {
   /**
-   * The deployment is usable; at least one node has the model allocated.
+   * The trained model is started on at least one node.
    */
   started,
   /**
-   * The deployment has recently started but is not yet usable; the model is not allocated on any nodes.
+   * Trained model deployment is starting but it is not yet deployed on any nodes.
    */
   starting,
   /**
-   * The deployment is preparing to stop and deallocate the model from the relevant nodes.
+   * Trained model deployment has started on all valid nodes.
    */
-  stopping
+  fully_allocated
 }
 
-export enum DeploymentAllocationState {
+export enum DeploymentAssignmentState {
   /**
-   * The trained model is started on at least one node.
+   * The deployment is usable; at least one node has the model allocated.
    */
   started,
   /**
-   * Trained model deployment is starting but it is not yet deployed on any nodes.
+   * The deployment has recently started but is not yet usable; the model is not allocated on any nodes.
    */
   starting,
   /**
-   * Trained model deployment has started on all valid nodes.
+   * The deployment is preparing to stop and deallocate the model from the relevant nodes.
    */
-  fully_allocated
-}
-
-export enum DeploymentAssignmentState {
-  starting,
-  started,
   stopping,
+  /**
+   * The deployment is on a failed state and must be re-deployed.
+   */
   failed
 }