diff --git a/monkey/tests/unit_tests/monkey_island/cc/repository/test_file_agent_configuration_repository.py b/monkey/tests/unit_tests/monkey_island/cc/repository/test_file_agent_configuration_repository.py
index 7f9f40d58..7782aa1ed 100644
--- a/monkey/tests/unit_tests/monkey_island/cc/repository/test_file_agent_configuration_repository.py
+++ b/monkey/tests/unit_tests/monkey_island/cc/repository/test_file_agent_configuration_repository.py
@@ -12,7 +12,7 @@ def repository(default_agent_configuration):
 
 
 def test_store_agent_config(repository):
-    agent_configuration = AgentConfiguration.from_mapping(AGENT_CONFIGURATION)
+    agent_configuration = AgentConfiguration(**AGENT_CONFIGURATION)
 
     repository.store_configuration(agent_configuration)
     retrieved_agent_configuration = repository.get_configuration()
@@ -36,7 +36,7 @@ def test_get_agent_config_retrieval_error(default_agent_configuration):
 
 
 def test_reset_to_default(repository, default_agent_configuration):
-    agent_configuration = AgentConfiguration.from_mapping(AGENT_CONFIGURATION)
+    agent_configuration = AgentConfiguration(**AGENT_CONFIGURATION)
 
     repository.store_configuration(agent_configuration)
     repository.reset_to_default()