diff --git a/tst/unit/env/envs/test_artifact_staging.py b/tst/unit/env/envs/test_artifact_staging.py
index 13cb6042..84dbf61b 100644
--- a/tst/unit/env/envs/test_artifact_staging.py
+++ b/tst/unit/env/envs/test_artifact_staging.py
@@ -14,7 +14,7 @@
 
 from agent_env.env.envs.mcp_server import MCPServerEnv
 from agent_env.env.envs.website import WebsiteEnv
-from agent_env.providers.sandbox_providers.local_sandbox import LocalSandbox
+from agent_env.providers.sandbox_providers.local_docker.provider import LocalSandbox
 
 _SHELL_NAMES = [
     ("email", "x;touch pwned;#.json"),