fix one code generation bug