diff --git a/smartsim/_core/mli/infrastructure/control/dragon_utils/dragon_util.py b/smartsim/_core/mli/infrastructure/control/dragon_util.py
similarity index 100%
rename from smartsim/_core/mli/infrastructure/control/dragon_utils/dragon_util.py
rename to smartsim/_core/mli/infrastructure/control/dragon_util.py
diff --git a/smartsim/_core/mli/infrastructure/control/dragon_utils/__init__.py b/smartsim/_core/mli/infrastructure/control/dragon_utils/__init__.py
deleted file mode 100644
index e69de29bb..000000000
diff --git a/smartsim/_core/mli/infrastructure/control/dragon_utils/msg_pump.py b/tests/dragon_wlm/msg_pump.py
similarity index 100%
rename from smartsim/_core/mli/infrastructure/control/dragon_utils/msg_pump.py
rename to tests/dragon_wlm/msg_pump.py
diff --git a/tests/dragon_wlm/test_request_dispatcher.py b/tests/dragon_wlm/test_request_dispatcher.py
index 07c6f7a49..3a2def8e3 100644
--- a/tests/dragon_wlm/test_request_dispatcher.py
+++ b/tests/dragon_wlm/test_request_dispatcher.py
@@ -53,12 +53,9 @@
 from smartsim._core.mli.comm.channel.dragon_channel import DragonCommChannel
 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.dragon_utils.dragon_util import (
+from smartsim._core.mli.infrastructure.control.dragon_util import (
     function_as_dragon_proc,
 )
-from smartsim._core.mli.infrastructure.control.dragon_utils.msg_pump import (
-    mock_messages,
-)
 from smartsim._core.mli.infrastructure.control.request_dispatcher import (
     RequestBatch,
     RequestDispatcher,
@@ -78,7 +75,8 @@
 from smartsim._core.mli.message_handler import MessageHandler
 from smartsim.log import get_logger
 
-from .utils.channel import FileSystemCommChannel
+from .channel import FileSystemCommChannel
+from .msg_pump import mock_messages
 
 logger = get_logger(__name__)