diff --git a/specification/cost-management/resource-manager/Microsoft.CostManagement/stable/2019-11-01/costmanagement.json b/specification/cost-management/resource-manager/Microsoft.CostManagement/stable/2019-11-01/costmanagement.json
index 2ace1a19a41d..8062bca713ac 100644
--- a/specification/cost-management/resource-manager/Microsoft.CostManagement/stable/2019-11-01/costmanagement.json
+++ b/specification/cost-management/resource-manager/Microsoft.CostManagement/stable/2019-11-01/costmanagement.json
@@ -2629,7 +2629,8 @@
       },
       "required": [
         "type",
-        "timeframe"
+        "timeframe",
+        "dataset"
       ]
     },
     "QueryDefinition": {
@@ -2675,7 +2676,8 @@
       },
       "required": [
         "type",
-        "timeframe"
+        "timeframe",
+        "dataset"
       ]
     },
     "QueryTimePeriod": {
diff --git a/specification/cost-management/resource-manager/Microsoft.CostManagement/stable/2020-06-01/costmanagement.json b/specification/cost-management/resource-manager/Microsoft.CostManagement/stable/2020-06-01/costmanagement.json
index 92d41577545c..3f8f4f596605 100644
--- a/specification/cost-management/resource-manager/Microsoft.CostManagement/stable/2020-06-01/costmanagement.json
+++ b/specification/cost-management/resource-manager/Microsoft.CostManagement/stable/2020-06-01/costmanagement.json
@@ -2159,7 +2159,8 @@
       },
       "required": [
         "type",
-        "timeframe"
+        "timeframe",
+        "dataset"
       ]
     },
     "ForecastDataset": {
@@ -2239,7 +2240,8 @@
       },
       "required": [
         "type",
-        "timeframe"
+        "timeframe",
+        "dataset"
       ]
     },
     "QueryTimePeriod": {