Codegen fixes