diff --git a/monkey/common/utils/exceptions.py b/monkey/common/utils/exceptions.py
index 2a0e369e9..945e35415 100644
--- a/monkey/common/utils/exceptions.py
+++ b/monkey/common/utils/exceptions.py
@@ -18,10 +18,6 @@ class IncorrectCredentialsError(Exception):
     """ Raise to indicate that authentication failed """
 
 
-class InvalidAWSKeys(Exception):
-    """ Raise to indicate that AWS API keys are invalid"""
-
-
 class NoInternetError(Exception):
     """ Raise to indicate problems caused when no internet connection is present"""
 
diff --git a/monkey/infection_monkey/exploit/caching_agent_repository.py b/monkey/infection_monkey/exploit/caching_agent_repository.py
index 8a55b3f63..9d24746ad 100644
--- a/monkey/infection_monkey/exploit/caching_agent_repository.py
+++ b/monkey/infection_monkey/exploit/caching_agent_repository.py
@@ -22,7 +22,7 @@ class CachingAgentRepository(IAgentRepository):
         self._proxies = proxies
         self._lock = threading.Lock()
 
-    def get_agent_binary(self, os: str, _: str = None) -> io.BytesIO:
+    def get_agent_binary(self, os: str, architecture: str = None) -> io.BytesIO:
         # If multiple calls to get_agent_binary() are made simultaneously before the result of
         # _download_binary_from_island() is cached, then multiple requests will be sent to the
         # island. Add a mutex in front of the call to _download_agent_binary_from_island() so