diff --git a/tests/unit/test_runtime_child_remap.py b/tests/unit/test_runtime_child_remap.py index 4bd3492..5c9a825 100644 --- a/tests/unit/test_runtime_child_remap.py +++ b/tests/unit/test_runtime_child_remap.py @@ -86,7 +86,7 @@ class RuntimeChildRemap(unittest.TestCase): before = run.download() old_allocations = list(allocations) if failure == 'construction': - operation = patch.object(module.compiler, 'compile_kernel', side_effect=RuntimeError('candidate compile')) + operation = patch.object(module.compiler, 'build_module', side_effect=RuntimeError('candidate compile')) else: initialize = type(run).initialize def fail(candidate, *arrays):