Add missing import to builder