diff --git a/tests/test_litellm/managed_agents/test_sessions.py b/tests/test_litellm/managed_agents/test_sessions.py
index eff52fa372e..3ab0d7b709f 100644
--- a/tests/test_litellm/managed_agents/test_sessions.py
+++ b/tests/test_litellm/managed_agents/test_sessions.py
@@ -28,7 +28,7 @@ from fastapi.testclient import TestClient
 
 sys.path.insert(0, os.path.abspath("../../.."))
 
-from litellm.managed_agents.endpoints import sessions as sessions_module
+from litellm.proxy.managed_agents_endpoints import sessions as sessions_module
 from litellm.proxy._types import CommonProxyErrors, UserAPIKeyAuth
 from litellm.proxy.auth.user_api_key_auth import user_api_key_auth