Make compliant for flake8