Fix flake error