Little pep8 fixes on the tests