Fix flake8 whitespace