diff --git a/litellm/tests/test_jwt.py b/litellm/tests/test_jwt.py
index 376b2bbe87..a2edbcc2f8 100644
--- a/litellm/tests/test_jwt.py
+++ b/litellm/tests/test_jwt.py
@@ -14,7 +14,7 @@ sys.path.insert(
 import pytest
 from litellm.proxy._types import LiteLLM_JWTAuth, LiteLLMRoutes
 from litellm.proxy.auth.handle_jwt import JWTHandler
-from litellm.proxy.team_endpoints import new_team
+from litellm.proxy.management_endpoints.team_endpoints import new_team
 from litellm.caching import DualCache
 from datetime import datetime, timedelta
 from fastapi import Request