codegen bug fix