fix flaky unit test