Updated script to fix change in codegen project