Move to flake8 and fix the warning