diff --git a/monkey/infection_monkey/exploit/tools.py b/monkey/infection_monkey/exploit/tools.py
index 31586417c..466834bcc 100644
--- a/monkey/infection_monkey/exploit/tools.py
+++ b/monkey/infection_monkey/exploit/tools.py
@@ -416,7 +416,7 @@ def get_interface_to_target(dst):
 
 
 def get_target_monkey(host):
-    from control import ControlClient
+    from infection_monkey.control import ControlClient
     import platform
     import sys
 
@@ -442,7 +442,7 @@ def get_target_monkey(host):
 
 
 def get_target_monkey_by_os(is_windows, is_32bit):
-    from control import ControlClient
+    from infection_monkey.control import ControlClient
     return ControlClient.download_monkey_exe_by_os(is_windows, is_32bit)