Fix pep8 test