fix code gen example