Fix style for flake8