Fixed INotifier functions regeneration
Added SegmentClass to generator attribute