fix flakiness