Overhaul defs to derive from a common base.

1 files changed, 2 insertions, 3 deletions

diff --git a/benchmarks/parsetostruct.upb_table.c b/benchmarks/parsetostruct.upb_table.c
index 4cf2b71..0a170de 100644
--- a/benchmarks/parsetostruct.upb_table.c
+++ b/benchmarks/parsetostruct.upb_table.c
@@ -31,15 +31,14 @@ static bool initialize()
   upb_string_unref(fds);
 
   struct upb_string *proto_name = upb_strdupc(MESSAGE_NAME);
-  struct upb_symtab_entry e;
-  if(!upb_context_lookup(c, proto_name, &e) || e.type != UPB_SYM_MESSAGE) {
+  def = upb_downcast_msgdef(upb_context_lookup(c, proto_name));
+  if(!def) {
     fprintf(stderr, "Error finding symbol '" UPB_STRFMT "'.\n",
             UPB_STRARG(proto_name));
     return false;
   }
   upb_string_unref(proto_name);
 
-  def = e.ref.msg;
   for(int i = 0; i < NUM_MESSAGES; i++)
     msgs[i] = upb_msg_new(def);