fix flaky tests 2: more flaky