Add missing import line