Fix flakiness.