[FIX] Flake8