diff --git a/tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_factory.py b/tests/test_litellm/managed_agents/vms/test_factory.py
similarity index 97%
rename from tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_factory.py
rename to tests/test_litellm/managed_agents/vms/test_factory.py
index 5b39a344288..2998c29b0fa 100644
--- a/tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_factory.py
+++ b/tests/test_litellm/managed_agents/vms/test_factory.py
@@ -11,7 +11,7 @@ from __future__ import annotations
 
 import pytest
 
-from litellm.proxy.agent_session_endpoints.vm_providers import (
+from litellm.managed_agents.vms import (
     EC2Provider,
     NoopProvider,
     SUPPORTED_PROVIDERS,