Add missing import.
Fix broken code introduced in b3a5095.