Fix a flake8 warning