Add missing import for Iterator