Fix flake8