Fix flake8 lint