diff --git a/litellm/proxy/agent_session_endpoints/sweepers.py b/litellm/proxy/agent_session_endpoints/sweepers.py
index 98b17149874..7cb2b517a74 100644
--- a/litellm/proxy/agent_session_endpoints/sweepers.py
+++ b/litellm/proxy/agent_session_endpoints/sweepers.py
@@ -25,7 +25,7 @@ from datetime import datetime, timedelta, timezone
 from typing import Any, Awaitable, Callable, Optional
 
 from litellm._logging import verbose_proxy_logger
-from litellm.proxy.agent_session_endpoints.vm_providers.base import (
+from litellm.managed_agents.vms.base import (
     AgentVMProvider,
     VMHandle,
 )