flake8 minor fix