diff --git a/tests/fake_lab/tasks/infinite_task.py b/tests/fake_lab/tasks/infinite_task.py index 92fbd5d4..3b3dcd5d 100644 --- a/tests/fake_lab/tasks/infinite_task.py +++ b/tests/fake_lab/tasks/infinite_task.py @@ -11,5 +11,6 @@ def __init__(self, samples: list[ObjectId], *args, **kwargs): self.sample = samples[0] def run(self): - with self.lab_view.request_resources({DeviceThatNeverEnds: {"infinite_loop": 1}}) as (device_that_fails, _): - device_that_fails.run_infinite() + with self.lab_view.request_resources({DeviceThatNeverEnds: {"infinite_loop": 1}}) as (devices, _): + device_that_never_ends = devices[DeviceThatNeverEnds] + device_that_never_ends.run_infinite()