fixing codegen