fix flake errors