diff --git a/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py b/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py
index e9bcbae62..cf97ac08a 100644
--- a/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py
+++ b/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py
@@ -6,10 +6,14 @@ from monkey_island.cc.repository import IAgentConfigurationRepository
 
 class InMemoryAgentConfigurationRepository(IAgentConfigurationRepository):
     def __init__(self):
-        self._configuration = AgentConfiguration.from_mapping(AGENT_CONFIGURATION)
+        self._default_configuration = AgentConfiguration.from_mapping(AGENT_CONFIGURATION)
+        self._configuration = self._default_configuration
 
     def get_configuration(self):
         return self._configuration
 
     def store_configuration(self, agent_configuration):
         self._configuration = agent_configuration
+
+    def reset_to_default(self):
+        self._configuration = self._default_configuration