[FIX] Fix Flake8 errors