Fix bug introduced when tidying for pep8