fix a flaky test