diff --git a/tests/test_litellm/proxy/agent_session_endpoints/test_sweepers.py b/tests/test_litellm/proxy/agent_session_endpoints/test_sweepers.py
index 22a94539abe..bf53c7fe30d 100644
--- a/tests/test_litellm/proxy/agent_session_endpoints/test_sweepers.py
+++ b/tests/test_litellm/proxy/agent_session_endpoints/test_sweepers.py
@@ -27,7 +27,7 @@ from litellm.proxy.agent_session_endpoints.sweepers import (
     STATUS_READY,
     STATUS_TERMINATED,
 )
-from litellm.proxy.agent_session_endpoints.vm_providers.base import (
+from litellm.managed_agents.vms.base import (
     AgentVMProvider,
     ProvisionContext,
     VMHandle,