diff --git a/monkey/infection_monkey/telemetry/messengers/telemetry_messenger_wrapper.py b/monkey/infection_monkey/telemetry/messengers/telemetry_messenger_wrapper.py
index d00f0dd23..e436cdd46 100644
--- a/monkey/infection_monkey/telemetry/messengers/telemetry_messenger_wrapper.py
+++ b/monkey/infection_monkey/telemetry/messengers/telemetry_messenger_wrapper.py
@@ -1,7 +1,7 @@
-from infection_monkey.telemetry.base_telem import BaseTelem
+from infection_monkey.telemetry.i_telem import ITelem
 from infection_monkey.telemetry.messengers.i_telemetry_messenger import ITelemetryMessenger
 
 
 class TelemetryMessengerWrapper(ITelemetryMessenger):
-    def send_telemetry(self, telemetry: BaseTelem):
+    def send_telemetry(self, telemetry: ITelem):
         telemetry.send()