diff --git a/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py b/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py
index 3c51297c5c3..bec62155f5f 100644
--- a/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py
+++ b/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py
@@ -5097,8 +5097,13 @@ async def test_resolve_tag_rate_limits_from_db_reads_budget_row(monkeypatch):
     from litellm.proxy import proxy_server
     from litellm.proxy.auth import auth_checks
     from litellm.proxy.hooks.parallel_request_limiter_v3 import resolve_tag_rate_limits_from_db
+    from litellm.proxy.utils import PrismaClient
 
-    async def fake_batch(tag_names, prisma_client, user_api_key_cache, **kwargs):
+    async def fake_batch(
+        tag_names: Sequence[str],
+        prisma_client: PrismaClient | None,
+        user_api_key_cache: DualCache,
+    ) -> dict[str, LiteLLM_TagTable]:
         return {
             "limited": LiteLLM_TagTable(
                 tag_name="limited",