fix CodeGenerator