diff --git a/features/steps/cancel_worktree_cleanup_steps.py b/features/steps/cancel_worktree_cleanup_steps.py
index 9845bbd3f..e736f1319 100644
--- a/features/steps/cancel_worktree_cleanup_steps.py
+++ b/features/steps/cancel_worktree_cleanup_steps.py
@@ -112,7 +112,7 @@ def step_call_cleanup(context: object, plan_id: str) -> None:
     from cleveragents.cli.commands.plan import _cleanup_sandbox_for_plan
 
     with patch(
-        "cleveragents.application.container.get_container",
+        "cleveragents.cli.commands.plan.get_container",
         return_value=context.cwc_container,
     ):
         _cleanup_sandbox_for_plan(plan_id, context.cwc_service)