Add missing import

Authored by bgamari on Tue, Jun 4, 5:18 PM.