Fixed flake8 nagging