diff --git a/tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py b/tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py
index a5691d49a64..ec352511d50 100644
--- a/tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py
+++ b/tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py
@@ -617,7 +617,8 @@ async def test_budgeted_invocation_requires_a_bounded_price(
     from litellm.proxy.agent_endpoints.auth.managed_authorization import prepare_agent_invocation
 
     budget: Final = {"budget_id": "budget", "max_budget": 1.0} if bounded else None
-    target: Final = agent(litellm_params=pricing, litellm_budget_table=budget)
+    target_budget: Final = {"budget_id": "target-budget", "max_budget": 1.0} if bounded != autonomous else None
+    target: Final = agent(litellm_params=pricing, litellm_budget_table=target_budget)
     registry: Final = agent_registry.AgentRegistry()
     registry.register_agent(target)
     database: Final = MagicMock()