Update to generator