diff --git a/tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_noop.py b/tests/test_litellm/managed_agents/vms/test_noop.py
similarity index 95%
rename from tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_noop.py
rename to tests/test_litellm/managed_agents/vms/test_noop.py
index 92d4de96833..eb73b85ce23 100644
--- a/tests/test_litellm/proxy/agent_session_endpoints/vm_providers/test_noop.py
+++ b/tests/test_litellm/managed_agents/vms/test_noop.py
@@ -4,7 +4,7 @@ from __future__ import annotations
 
 import pytest
 
-from litellm.proxy.agent_session_endpoints.vm_providers import (
+from litellm.managed_agents.vms import (
     NoopProvider,
     ProvisionContext,
     VMState,