diff --git a/output/schema/schema.json b/output/schema/schema.json
index 9fd29666de..abd1be0ee8 100644
--- a/output/schema/schema.json
+++ b/output/schema/schema.json
@@ -153254,10 +153254,13 @@
           "name": "nodes",
           "required": true,
           "type": {
-            "kind": "instance_of",
-            "type": {
-              "name": "TrainedModelDeploymentNodesStats",
-              "namespace": "ml._types"
+            "kind": "array_of",
+            "value": {
+              "kind": "instance_of",
+              "type": {
+                "name": "TrainedModelDeploymentNodesStats",
+                "namespace": "ml._types"
+              }
             }
           }
         },
@@ -153574,9 +153577,18 @@
           "name": "timestamp",
           "required": true,
           "type": {
+            "generics": [
+              {
+                "kind": "instance_of",
+                "type": {
+                  "name": "UnitMillis",
+                  "namespace": "_types"
+                }
+              }
+            ],
             "kind": "instance_of",
             "type": {
-              "name": "DateTime",
+              "name": "EpochTime",
               "namespace": "_types"
             }
           }
diff --git a/output/typescript/types.ts b/output/typescript/types.ts
index 0efd628c2d..d788226fcd 100644
--- a/output/typescript/types.ts
+++ b/output/typescript/types.ts
@@ -14082,7 +14082,7 @@ export interface MlTrainedModelDeploymentStats {
   error_count: integer
   inference_count: integer
   model_id: Id
-  nodes: MlTrainedModelDeploymentNodesStats
+  nodes: MlTrainedModelDeploymentNodesStats[]
   number_of_allocations: integer
   queue_capacity: integer
   rejected_execution_count: integer
@@ -14117,7 +14117,7 @@ export interface MlTrainedModelInferenceStats {
   failure_count: integer
   inference_count: integer
   missing_all_fields_count: integer
-  timestamp: DateTime
+  timestamp: EpochTime<UnitMillis>
 }
 
 export interface MlTrainedModelLocation {
diff --git a/specification/ml/_types/TrainedModel.ts b/specification/ml/_types/TrainedModel.ts
index 2376b9fe69..d8b6249c69 100644
--- a/specification/ml/_types/TrainedModel.ts
+++ b/specification/ml/_types/TrainedModel.ts
@@ -75,7 +75,7 @@ export class TrainedModelDeploymentStats {
    * The deployment stats for each node that currently has the model allocated.
    * In serverless, stats are reported for a single unnamed virtual node.
    */
-  nodes: TrainedModelDeploymentNodesStats
+  nodes: TrainedModelDeploymentNodesStats[]
   /** The number of allocations requested. */
   number_of_allocations: integer
   /** The number of inference requests that can be queued before new requests are rejected. */
@@ -120,7 +120,7 @@ export class TrainedModelInferenceStats {
   /** The number of inference calls where all the training features for the model were missing. */
   missing_all_fields_count: integer
   /** The time when the statistics were last updated. */
-  timestamp: DateTime
+  timestamp: EpochTime<UnitMillis>
 }
 
 export class TrainedModelSizeStats {