Revert import to original in example.

This was changed to make the example code run with my changes on a local branch; it needs to be reverted for the pull request. Found by dgryski.