Fix flake8 err