fix flaky test