Fix a bug introduces by pep8 cleanup.