Add missing import statement

refactoring-induced missing import.

Signed-off-by: Joe Buck <jbbuck@gmail.com>