[FIX] flake8 errors