diff --git a/monkey/infection_monkey/island_api_client/i_island_api_client.py b/monkey/infection_monkey/island_api_client/i_island_api_client.py
index d14c8deeb..86d42ee88 100644
--- a/monkey/infection_monkey/island_api_client/i_island_api_client.py
+++ b/monkey/infection_monkey/island_api_client/i_island_api_client.py
@@ -5,7 +5,7 @@ from common import AgentRegistrationData, AgentSignals, OperatingSystem
 from common.agent_configuration import AgentConfiguration
 from common.agent_events import AbstractAgentEvent
 from common.credentials import Credentials
-from common.types import SocketAddress
+from common.types import AgentID, SocketAddress
 
 
 class IIslandAPIClient(ABC):
@@ -30,7 +30,7 @@ class IIslandAPIClient(ABC):
         """
 
     @abstractmethod
-    def send_log(self, agent_id: int, log_contents: str):
+    def send_log(self, agent_id: AgentID, log_contents: str):
         """
         Send the contents of the agent's log to the island