Fix flake8 issue