diff --git a/dask_cuda/tests/test_dask_cuda_worker.py b/dask_cuda/tests/test_dask_cuda_worker.py
index 6f653a36..ed47aefb 100644
--- a/dask_cuda/tests/test_dask_cuda_worker.py
+++ b/dask_cuda/tests/test_dask_cuda_worker.py
@@ -10,8 +10,7 @@
 
 from distributed import Client, wait
 from distributed.system import MEMORY_LIMIT
-from distributed.utils_test import loop  # noqa: F401
-from distributed.utils_test import popen
+from distributed.utils_test import cleanup, loop, popen  # noqa: F401
 
 from dask_cuda.utils import (
     get_gpu_count_mig,