diff --git a/monkey/infection_monkey/master/automated_master.py b/monkey/infection_monkey/master/automated_master.py
index 3a4ef6835..88841f401 100644
--- a/monkey/infection_monkey/master/automated_master.py
+++ b/monkey/infection_monkey/master/automated_master.py
@@ -9,13 +9,13 @@ from infection_monkey.i_master import IMaster
 from infection_monkey.i_puppet import IPuppet
 from infection_monkey.model import VictimHostFactory
 from infection_monkey.network import NetworkInterface
+from infection_monkey.post_breach.custom_pba import CustomPBA
 from infection_monkey.telemetry.credentials_telem import CredentialsTelem
 from infection_monkey.telemetry.messengers.i_telemetry_messenger import ITelemetryMessenger
 from infection_monkey.telemetry.post_breach_telem import PostBreachTelem
 from infection_monkey.utils.threading import create_daemon_thread, interruptible_iter
 from infection_monkey.utils.timer import Timer
 
-from ..post_breach.custom_pba.custom_pba import CustomPBA
 from . import Exploiter, IPScanner, Propagator
 
 CHECK_ISLAND_FOR_STOP_COMMAND_INTERVAL_SEC = 5
diff --git a/monkey/infection_monkey/monkey.py b/monkey/infection_monkey/monkey.py
index b500fa27f..96f1873bc 100644
--- a/monkey/infection_monkey/monkey.py
+++ b/monkey/infection_monkey/monkey.py
@@ -51,7 +51,7 @@ from infection_monkey.post_breach.actions.schedule_jobs import ScheduleJobs
 from infection_monkey.post_breach.actions.timestomping import Timestomping
 from infection_monkey.post_breach.actions.use_signed_scripts import SignedScriptProxyExecution
 from infection_monkey.post_breach.actions.use_trap_command import TrapCommand
-from infection_monkey.post_breach.custom_pba.custom_pba import CustomPBA
+from infection_monkey.post_breach.custom_pba import CustomPBA
 from infection_monkey.puppet.puppet import Puppet
 from infection_monkey.system_singleton import SystemSingleton
 from infection_monkey.telemetry.attack.t1106_telem import T1106Telem
diff --git a/monkey/infection_monkey/post_breach/custom_pba/__init__.py b/monkey/infection_monkey/post_breach/custom_pba/__init__.py
index e69de29bb..723810fc0 100644
--- a/monkey/infection_monkey/post_breach/custom_pba/__init__.py
+++ b/monkey/infection_monkey/post_breach/custom_pba/__init__.py
@@ -0,0 +1 @@
+from .custom_pba import CustomPBA