Fix flake8 warning