updare custom generator reference