Fix missing import.
Merge pull request #30 from JeremyRand/import_traceback

Fix missing import.