diff --git a/litellm/proxy/agent_session_endpoints/vm_providers/ec2.py b/litellm/managed_agents/vms/ec2.py
similarity index 99%
rename from litellm/proxy/agent_session_endpoints/vm_providers/ec2.py
rename to litellm/managed_agents/vms/ec2.py
index c3553e61e25..36c8105fc9e 100644
--- a/litellm/proxy/agent_session_endpoints/vm_providers/ec2.py
+++ b/litellm/managed_agents/vms/ec2.py
@@ -25,7 +25,7 @@ from concurrent.futures import ThreadPoolExecutor
 from typing import Any, Dict, List, 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,
     AwsCreds,
     Ec2Config,