Fixed codegen bug