Adds back an import that was no longer unused.

It was unused in the original base of unused-code, but became used by current
main.  This was causing import errors.