Revert "fix a traceback"

This reverts commit 48f7a1a441be2cbaba8b795d99ef40adab90ecf6.