Fix flake8 error