Fix a flaky test.