diff --git a/x-pack/plugin/src/test/resources/rest-api-spec/test/ml/explain_data_frame_analytics.yml b/x-pack/plugin/src/test/resources/rest-api-spec/test/ml/explain_data_frame_analytics.yml
index d567003313b..1a7a51dff3d 100644
--- a/x-pack/plugin/src/test/resources/rest-api-spec/test/ml/explain_data_frame_analytics.yml
+++ b/x-pack/plugin/src/test/resources/rest-api-spec/test/ml/explain_data_frame_analytics.yml
@@ -126,9 +126,9 @@
           source: { index: "index-source" }
           analysis: { outlier_detection: {} }
   - match:
-      memory_estimation.expected_memory_without_disk: / \dkb /
+      memory_estimation.expected_memory_without_disk: / \d[km]b /
   - match:
-      memory_estimation.expected_memory_with_disk: / \dkb /
+      memory_estimation.expected_memory_with_disk: / \d[km]b /
   - length: { field_selection: 2 }
   - match: { field_selection.0.name: "x" }
   - match: { field_selection.0.mapping_types: ["float"] }
@@ -156,9 +156,9 @@
           source: { index: "index-source" }
           analysis: { outlier_detection: {} }
   - match:
-      memory_estimation.expected_memory_without_disk: / \dkb /
+      memory_estimation.expected_memory_without_disk: / \d[km]b /
   - match:
-      memory_estimation.expected_memory_with_disk: / \dkb /
+      memory_estimation.expected_memory_with_disk: / \d[km]b /
 
   - do:
       index:
@@ -173,9 +173,9 @@
           source: { index: "index-source" }
           analysis: { outlier_detection: {} }
   - match:
-      memory_estimation.expected_memory_without_disk: / \dkb /
+      memory_estimation.expected_memory_without_disk: / \d[km]b /
   - match:
-      memory_estimation.expected_memory_with_disk: / \dkb /
+      memory_estimation.expected_memory_with_disk: / \d[km]b /
 
 ---
 "Test field_selection given body":