Fix flake8 output