Fix broken test
pep8