diff --git a/src/dask_awkward/lib/core.py b/src/dask_awkward/lib/core.py
index a9567a56..be2746b7 100644
--- a/src/dask_awkward/lib/core.py
+++ b/src/dask_awkward/lib/core.py
@@ -62,7 +62,7 @@
 )
 
 if _dask_uses_tasks:
-    from dask._task_spec import TaskRef
+    from dask.blockwise import TaskRef # type: ignore
 
 if TYPE_CHECKING:
     from awkward.contents.content import Content