Fixing for flake8