diff --git a/litellm/main.py b/litellm/main.py
index 2c906e990..6ad59dde8 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -14,7 +14,6 @@ from functools import partial
 import dotenv, traceback, random, asyncio, time, contextvars
 from copy import deepcopy
 import httpx
-
 import litellm
 from ._logging import verbose_logger
 from litellm import (  # type: ignore
diff --git a/tests/test_team.py b/tests/test_team.py
index 467767be0..544273c2e 100644
--- a/tests/test_team.py
+++ b/tests/test_team.py
@@ -578,4 +578,4 @@ async def test_users_in_team_budget():
         except Exception as e:
             print("got exception, this is expected")
             print(e)
-            assert "Crossed spend within team" in str(e)
+            assert "Budget has been exceeded" in str(e)