Fixing code generation bug.