Fix flake errors