Fixing for flake8
Fixing