Fixing test/pep8