diff --git a/tests/test_full_stack_contract.py b/tests/test_full_stack_contract.py
index e41be6d..442fa30 100644
--- a/tests/test_full_stack_contract.py
+++ b/tests/test_full_stack_contract.py
@@ -16,6 +16,7 @@ def load_agent_module():
     spec = importlib.util.spec_from_file_location("contract_clock_agent", ROOT / "agent.py")
     module = importlib.util.module_from_spec(spec)
     assert spec.loader is not None
+    sys.modules[spec.name] = module
     spec.loader.exec_module(module)
     return module