Add missing import reference