make unit test less flaky