diff --git a/monkey/monkey_island/cc/resources/__init__.py b/monkey/monkey_island/cc/resources/__init__.py
index 3368869af..d04190b7c 100644
--- a/monkey/monkey_island/cc/resources/__init__.py
+++ b/monkey/monkey_island/cc/resources/__init__.py
@@ -6,5 +6,5 @@ from .reset_agent_configuration import ResetAgentConfiguration
 from .propagation_credentials import PropagationCredentials
 from .ip_addresses import IPAddresses
 from .agent_configuration import AgentConfiguration
-from .pba_file_upload import PBAFileUpload
+from .pba_file_upload import PBAFileUpload, LINUX_PBA_TYPE, WINDOWS_PBA_TYPE
 from .pba_file_download import PBAFileDownload
diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py b/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py
index 397a5f6ab..a40eb60f8 100644
--- a/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py
+++ b/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py
@@ -9,8 +9,7 @@ from tests.utils import raise_
 
 from common import DIContainer
 from monkey_island.cc.repository import IAgentConfigurationRepository, IFileRepository
-from monkey_island.cc.resources import PBAFileUpload
-from monkey_island.cc.resources.pba_file_upload import LINUX_PBA_TYPE, WINDOWS_PBA_TYPE
+from monkey_island.cc.resources import LINUX_PBA_TYPE, WINDOWS_PBA_TYPE, PBAFileUpload
 
 TEST_FILE_CONTENTS = b"m0nk3y"
 TEST_FILE = (