Fix traceback being lost

This fixes #158.