Fix the dependency on flake