Flake8 fixes
Another bug