From 0a9d221ad8cf79e5c82c510aed4085aa863547ac Mon Sep 17 00:00:00 2001
From: Kekoa Kaaikala <kekoa.kaaikala@gmail.com>
Date: Wed, 21 Sep 2022 17:47:11 +0000
Subject: [PATCH] Agent: Fix mypy issues in HostExploiter.py

---
 monkey/infection_monkey/exploit/HostExploiter.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/monkey/infection_monkey/exploit/HostExploiter.py b/monkey/infection_monkey/exploit/HostExploiter.py
index 6c7128677..6291d98d3 100644
--- a/monkey/infection_monkey/exploit/HostExploiter.py
+++ b/monkey/infection_monkey/exploit/HostExploiter.py
@@ -7,6 +7,7 @@ from typing import Dict, Sequence
 from common.event_queue import IAgentEventQueue
 from common.utils.exceptions import FailedExploitationError
 from infection_monkey.i_puppet import ExploiterResultData
+from infection_monkey.model import VictimHost
 from infection_monkey.telemetry.messengers.i_telemetry_messenger import ITelemetryMessenger
 
 from . import IAgentBinaryRepository
@@ -58,7 +59,7 @@ class HostExploiter:
     # TODO: host should be VictimHost, at the moment it can't because of circular dependency
     def exploit_host(
         self,
-        host,
+        host: VictimHost,
         servers: Sequence[str],
         current_depth: int,
         telemetry_messenger: ITelemetryMessenger,