Added missing import in builder