diff --git a/docs/src/test/java/org/elasticsearch/smoketest/DocsClientYamlTestSuiteIT.java b/docs/src/test/java/org/elasticsearch/smoketest/DocsClientYamlTestSuiteIT.java
index ffdabd6d63399..14fef43344644 100644
--- a/docs/src/test/java/org/elasticsearch/smoketest/DocsClientYamlTestSuiteIT.java
+++ b/docs/src/test/java/org/elasticsearch/smoketest/DocsClientYamlTestSuiteIT.java
@@ -101,7 +101,7 @@ protected ClientYamlTestClient initClientYamlTestClient(
 
     @After
     public void cleanup() throws Exception {
-        if (isMachineLearningTest() || isDataFrameTest()) {
+        if (isMachineLearningTest() || isTransformTest()) {
             ESRestTestCase.waitForPendingTasks(adminClient());
         }
     }
@@ -111,9 +111,9 @@ protected boolean isMachineLearningTest() {
         return testName != null && (testName.contains("/ml/") || testName.contains("\\ml\\"));
     }
 
-    protected boolean isDataFrameTest() {
+    protected boolean isTransformTest() {
         String testName = getTestName();
-        return testName != null && (testName.contains("/data-frames/") || testName.contains("\\data-frames\\"));
+        return testName != null && (testName.contains("/transform/") || testName.contains("\\transform\\"));
     }
 
     /**