Fixing flake8 issue