[FIX] Fix problems with flake8