diff --git a/distributed/scheduler.py b/distributed/scheduler.py
index 8911e72247..877f766274 100644
--- a/distributed/scheduler.py
+++ b/distributed/scheduler.py
@@ -3082,7 +3082,7 @@ def is_rootish(self, ts: TaskState) -> bool:
         to override this heuristic.
         """
         # Check explicitly marked IO tasks
-        if ts.run_spec and ts.run_spec.io_task:
+        if ts.run_spec and ts.run_spec.data_producer_task:
             return True
 
         if ts.resource_restrictions or ts.worker_restrictions or ts.host_restrictions: