From 0ab0111d57d26b6f9bcb29c1f6d74a56c6f97bd1 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 11 Jun 2024 18:05:20 -0700
Subject: [PATCH] test(test_team.py): fix error string asserted

---
 litellm/main.py    | 1 -
 tests/test_team.py | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

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)