bug fix for codegen algo