Fixed an import error caused by refactoring.

Signed-off-by: Jared Woolston <jwoolston@tenkiv.com>