[FIX] Test Flake8