diff --git a/tests/protocol_test.py b/tests/protocol_test.py
index 099e1cb..4d260be 100644
--- a/tests/protocol_test.py
+++ b/tests/protocol_test.py
@@ -105,7 +105,7 @@ async def test_AIQueryApi_create_task(ai_query_api_build_fixture, ai_query_api_s
     assert agent.did.startswith("did:")
     print('Agent service created:', agent.did)
 
-    await builder.ai_protocol.subscribe(eventsReceived, agent.did, ['step-updated'])
+    await builder.ai_protocol.subscribe(eventsReceived)
     assert builder.ai_protocol.socket_client.connected
     assert builder.room_id