pep8 fixes tests