Add missing import call