diff --git a/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py b/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py index 5eb8f0310..0d1f4127c 100644 --- a/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py +++ b/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py @@ -9,7 +9,7 @@ from infection_monkey.tunnel import MonkeyTunnel class ExploitInterceptingTelemetryMessenger(ITelemetryMessenger): def __init__( - self, telemetry_messenger: ITelemetryMessenger, tunnel: MonkeyTunnel, relay: TCPRelay = None + self, telemetry_messenger: ITelemetryMessenger, tunnel: MonkeyTunnel, relay: TCPRelay ): self._telemetry_messenger = telemetry_messenger self._tunnel = tunnel @@ -26,7 +26,7 @@ def _send_telemetry( telemetry: ITelem, telemetry_messenger: ITelemetryMessenger, tunnel: MonkeyTunnel, - relay: TCPRelay = None, + relay: TCPRelay, ): telemetry_messenger.send_telemetry(telemetry) @@ -36,7 +36,7 @@ def _( telemetry: ExploitTelem, telemetry_messenger: ITelemetryMessenger, tunnel: MonkeyTunnel, - relay: TCPRelay = None, + relay: TCPRelay, ): if telemetry.propagation_result is True: tunnel.set_wait_for_exploited_machines()