Fix line length for flake8