[FIX] Flake8 minor thing