Fixing codegen.