pep8: fix tests after pep8 changes