Optimize UnionTypeCodec