minor update to generator