Fix flake8 pragma