diff --git a/monkey/infection_monkey/network/relay/__init__.py b/monkey/infection_monkey/network/relay/__init__.py
index 737d5baf1..ad5d325c5 100644
--- a/monkey/infection_monkey/network/relay/__init__.py
+++ b/monkey/infection_monkey/network/relay/__init__.py
@@ -1,4 +1,4 @@
-from .relay_connection_handler import RelayConnectionHandler
+from .relay_connection_handler import RelayConnectionHandler, RELAY_CONTROL_MESSAGE
 from .relay_user_handler import RelayUser, RelayUserHandler
 from .sockets_pipe import SocketsPipe
 from .tcp_connection_handler import TCPConnectionHandler
diff --git a/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py b/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py
index c871681b6..f82432dae 100644
--- a/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py
+++ b/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py
@@ -5,11 +5,11 @@ from unittest.mock import MagicMock
 import pytest
 
 from monkey.infection_monkey.network.relay import (
+    RELAY_CONTROL_MESSAGE,
     RelayConnectionHandler,
     RelayUserHandler,
     TCPPipeSpawner,
 )
-from monkey.infection_monkey.network.relay.relay_connection_handler import RELAY_CONTROL_MESSAGE
 
 USER_ADDRESS = "0.0.0.1"