diff --git a/tests/dragon/test_worker_manager.py b/tests/dragon/test_worker_manager.py
index 0feefdb510..69d962940b 100644
--- a/tests/dragon/test_worker_manager.py
+++ b/tests/dragon/test_worker_manager.py
@@ -31,8 +31,6 @@
 
 import pytest
 
-from smartsim._core.mli.comm.channel.dragon_util import create_local
-
 torch = pytest.importorskip("torch")
 dragon = pytest.importorskip("dragon")
 
@@ -56,6 +54,7 @@
 from dragon.data.ddict.ddict import DDict
 
 from smartsim._core.mli.comm.channel.dragon_fli import DragonFLIChannel
+from smartsim._core.mli.comm.channel.dragon_util import create_local
 from smartsim._core.mli.infrastructure.control.worker_manager import (
     EnvironmentConfigLoader,
     WorkerManager,