[REF] Fix flake8