Correct flake8 errors