Update for latest flake8