Fix failing test due to previous Pep8 change