diff --git a/tests/conftest.py b/tests/conftest.py
index 7cca8ee..4e62f32 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -8,7 +8,7 @@
 from agent import ip2geo_agent
 
 @pytest.fixture(scope='function', name='ip2geo_agent')
-def fixture_io2geo_agent():
+def fixture_io2geo_agent(agent_mock):
     with (pathlib.Path(__file__).parent.parent / 'ostorlab.yaml').open() as yaml_o:
         definition = agent_definitions.AgentDefinition.from_yaml(yaml_o)
         settings = runtime_definitions.AgentSettings(