diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py
index 47a2ff6dc..effac5890 100644
--- a/litellm/tests/test_key_generate_prisma.py
+++ b/litellm/tests/test_key_generate_prisma.py
@@ -664,7 +664,7 @@ def test_call_with_key_over_budget(prisma_client):
         asyncio.run(test())
     except Exception as e:
         error_detail = e.detail
-        assert "Authentication Error, ExceededBudget:" in error_detail
+        assert "Authentication Error, ExceededTokenBudget:" in error_detail
         print(vars(e))