[FIX] Fixed errors that make flake8 fail