Fix flake8 syntax