FIX flake8