diff --git a/robot/helper_plan_correct_isolated_resolve.py b/robot/helper_plan_correct_isolated_resolve.py index 7a770cb77..13b15f71f 100644 --- a/robot/helper_plan_correct_isolated_resolve.py +++ b/robot/helper_plan_correct_isolated_resolve.py @@ -13,7 +13,7 @@ import os import tempfile from pathlib import Path from types import SimpleNamespace -from unittest.mock import MagicMock, patch +from unittest.mock import MagicMock from typer.testing import CliRunner from ulid import ULID @@ -90,10 +90,8 @@ def run_regression() -> int: ) runner = CliRunner() - with patch( - "cleveragents.application.services.correction_service.CorrectionService", - return_value=fake_svc, - ): + container = get_container() + with container.correction_service.override(fake_svc): result = runner.invoke( plan_app, [