diff --git a/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py b/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py
index 3f9f2bacdd3f..0c362db88532 100644
--- a/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py
+++ b/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py
@@ -333,6 +333,7 @@ async def fake_async_vector_store_search_handler(
         timeout=None,
         client=None,
         _is_async=False,
+        router: "litellm.Router | None" = None,
     ):
         litellm_params_dict = (
             litellm_params.model_dump(exclude_none=False)