diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/restartComputeInstance.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/restartComputeInstance.json
index 7bb50a4f6560..d7240808c50c 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/restartComputeInstance.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/restartComputeInstance.json
@@ -7,6 +7,6 @@
     "api-version": "2021-01-01"
   },
   "responses": {
-    "200": {}
+    "202": {}
   }
 }
diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/machineLearningServices.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/machineLearningServices.json
index 2e263b6c0bec..6c701fbd1e2c 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/machineLearningServices.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/machineLearningServices.json
@@ -1180,6 +1180,7 @@
             "$ref": "./examples/restartComputeInstance.json"
           }
         },
+        "x-ms-long-running-operation": true,
         "parameters": [
           {
             "$ref": "#/parameters/SubscriptionIdParameter"
@@ -1198,7 +1199,7 @@
           }
         ],
         "responses": {
-          "200": {
+          "202": {
             "description": "Success."
           },
           "default": {