fix flaky tests