diff --git a/src/dask_awkward/layers/layers.py b/src/dask_awkward/layers/layers.py
index 305409a2..27e1f15d 100644
--- a/src/dask_awkward/layers/layers.py
+++ b/src/dask_awkward/layers/layers.py
@@ -12,10 +12,10 @@
 
 from dask_awkward.utils import LazyInputsDict
 
-_dask_uses_tasks = hasattr(dask, "_task_spec") and hasattr(dask.blockwise, "Task")
+_dask_uses_tasks = hasattr(dask.blockwise, "Task")
 
 if _dask_uses_tasks:
-    from dask._task_spec import Task, TaskRef
+    from dask.blockwise import Task, TaskRef
 
 if TYPE_CHECKING:
     from awkward import Array as AwkwardArray