Fix bug introduced by pep8 changes