Fixed failing codegen test