diff --git a/tests/test_agent.py b/tests/test_agent.py
index 77823c6..aaf2155 100644
--- a/tests/test_agent.py
+++ b/tests/test_agent.py
@@ -137,6 +137,9 @@ def __init__(self):
         self.reserve_pending = asyncio.Event()
         self.reserve_continue.set()
 
+    def set_api_object(self, _):
+        pass
+
     async def reserve(self, candidate_ids):
         self.reserve_pending.set()
         try: