diff --git a/packages/service-library/tests/deferred_tasks/test__base_deferred_handler.py b/packages/service-library/tests/deferred_tasks/test__base_deferred_handler.py index 80a6090a63e..9f3451058bf 100644 --- a/packages/service-library/tests/deferred_tasks/test__base_deferred_handler.py +++ b/packages/service-library/tests/deferred_tasks/test__base_deferred_handler.py @@ -445,7 +445,7 @@ async def _run_that_times_out(_: DeferredContext) -> None: await _assert_mock_call(mocks, key=MockKeys.ON_FINISHED_WITH_ERROR, count=1) for entry in mocks[MockKeys.ON_FINISHED_WITH_ERROR].call_args_list: - assert "asyncio.exceptions.TimeoutError" in entry.args[0].error + assert "builtins.TimeoutError" in entry.args[0].error await _assert_mock_call(mocks, key=MockKeys.RUN_DEFERRED, count=0) await _assert_mock_call(mocks, key=MockKeys.ON_DEFERRED_RESULT, count=0)