Fix flaky test.