Fix a unit test (random failures)