diff --git a/tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_ec2_provider.py b/tests/test_litellm/managed_agents/vms/test_ec2_provider.py
similarity index 99%
rename from tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_ec2_provider.py
rename to tests/test_litellm/managed_agents/vms/test_ec2_provider.py
index d260e20625a..d45381770c7 100644
--- a/tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_ec2_provider.py
+++ b/tests/test_litellm/managed_agents/vms/test_ec2_provider.py
@@ -20,7 +20,7 @@ from unittest.mock import patch
 
 import pytest
 
-from litellm.proxy.agent_session_endpoints.vm_providers import (
+from litellm.managed_agents.vms import (
     AwsCreds,
     EC2Provider,
     Ec2Config,