diff --git a/src/dask_awkward/lib/optimize.py b/src/dask_awkward/lib/optimize.py
index 4b9dd6cf..0eba92fa 100644
--- a/src/dask_awkward/lib/optimize.py
+++ b/src/dask_awkward/lib/optimize.py
@@ -22,7 +22,7 @@
 from dask_awkward.utils import first
 
 if _dask_uses_tasks:
-    from dask._task_spec import GraphNode, Task, TaskRef
+    from dask.blockwise import GraphNode, Task, TaskRef
 
 if TYPE_CHECKING:
     from awkward._nplikes.typetracer import TypeTracerReport