diff --git a/tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_ec2_provider_real.py b/tests/test_litellm/managed_agents/vms/test_ec2_provider_real.py
similarity index 98%
rename from tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_ec2_provider_real.py
rename to tests/test_litellm/managed_agents/vms/test_ec2_provider_real.py
index e99b01f1515..658383a4a70 100644
--- a/tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_ec2_provider_real.py
+++ b/tests/test_litellm/managed_agents/vms/test_ec2_provider_real.py
@@ -30,7 +30,7 @@ from typing import List, Optional
 
 import pytest
 
-from litellm.proxy.agent_session_endpoints.vm_providers import (
+from litellm.managed_agents.vms import (
     AwsCreds,
     EC2Provider,
     Ec2Config,
@@ -153,7 +153,7 @@ async def test_real_boot():
 @pytest.mark.asyncio
 async def test_byoc_invalid_creds_fail_fast():
     """Validation #11: bad creds → InvalidCredentialsError within ~5s, no instance launched."""
-    from litellm.proxy.agent_session_endpoints.vm_providers import (
+    from litellm.managed_agents.vms import (
         InvalidCredentialsError,
     )