diff --git a/gradle/dump_hanging_test.gradle b/gradle/dump_hanging_test.gradle
index 3a322463ed2..d10ea461b44 100644
--- a/gradle/dump_hanging_test.gradle
+++ b/gradle/dump_hanging_test.gradle
@@ -68,16 +68,16 @@ tasks.withType(Test).configureEach { testTask ->
     }, delayMinutes, TimeUnit.MINUTES)
 
     // Store handles for cancellation in doLast.
-    ext.dumpFuture = future
-    ext.dumpScheduler = scheduler
+    testTask.ext.dumpFuture = future
+    testTask.ext.dumpScheduler = scheduler
   }
 
   doLast {
     // Cancel if the task finished before the scheduled dump.
     try {
-      ext.dumpFuture?.cancel(false)
+      testTask.ext.dumpFuture?.cancel(false)
     } finally {
-      ext.dumpScheduler?.shutdownNow()
+      testTask.ext.dumpScheduler?.shutdownNow()
     }
   }
 }