Fix flakes