Update test to reflect fix in pep8