Fixing pep8 test