diff --git a/tests/test_litellm/proxy/auth/test_auth_checks.py b/tests/test_litellm/proxy/auth/test_auth_checks.py
index 582327b3ad0..ee316933e68 100644
--- a/tests/test_litellm/proxy/auth/test_auth_checks.py
+++ b/tests/test_litellm/proxy/auth/test_auth_checks.py
@@ -8045,7 +8045,7 @@ async def test_check_team_member_budget_missing_membership_is_verified_with_unav
         patch.object(proxy_server, "prisma_client", prisma_client),
         patch.object(proxy_server, "spend_counter_cache", _unavailable_spend_counter_cache()),
     ):
-        await _check_team_member_budget(
+        result: Final = await _check_team_member_budget(
             team_object=LiteLLM_TeamTable(
                 team_id="test-team",
                 metadata={"team_member_budget_id": "default-budget-100"},
@@ -8059,6 +8059,7 @@ async def test_check_team_member_budget_missing_membership_is_verified_with_unav
             team_membership_loaded=True,
         )
 
+    assert result is None
     membership_find_unique.assert_awaited()
 
 
diff --git a/tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py b/tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py
index 922504ecc58..64f9b3b1540 100644
--- a/tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py
+++ b/tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py
@@ -884,7 +884,14 @@ async def test_team_update_reaches_inherited_members_but_not_overridden_ones():
             team_member_budget=1.0,
         )
 
-    async def spend_from_membership(counter_key: str, fallback_spend: float, max_budget: float | None = None) -> float:
+    async def spend_from_membership(
+        counter_key: str,
+        fallback_spend: float,
+        max_budget: float | None = None,
+        *,
+        fallback_authoritative: bool,
+    ) -> float:
+        assert not fallback_authoritative
         return fallback_spend
 
     async def check(user_id: str, spend: float) -> None: