diff --git a/.isort.cfg b/.isort.cfg
index 3f01d0d..e390adf 100644
--- a/.isort.cfg
+++ b/.isort.cfg
@@ -1,2 +1,3 @@
 [settings]
 known_third_party = click,dask,distributed,mpi4py,pytest,requests,setuptools,yaml
+profile = black
diff --git a/dask_mpi/tests/test_cli.py b/dask_mpi/tests/test_cli.py
index 6e6dcdf..5befa50 100644
--- a/dask_mpi/tests/test_cli.py
+++ b/dask_mpi/tests/test_cli.py
@@ -12,8 +12,7 @@
 from distributed.comm.addressing import get_address_host_port
 from distributed.metrics import time
 from distributed.utils import import_term, tmpfile
-from distributed.utils_test import loop  # noqa: F401
-from distributed.utils_test import popen
+from distributed.utils_test import cleanup, loop, loop_in_thread, popen  # noqa: F401
 
 pytest.importorskip("mpi4py")