Fixing flake8 error