diff --git a/Driver/RewriteTest.cpp b/Driver/RewriteTest.cpp
index 1cf5895..3904d86 100644
--- a/Driver/RewriteTest.cpp
+++ b/Driver/RewriteTest.cpp
@@ -686,7 +686,11 @@
            CDecl->getName());
   else
     printf(", 0\n");
-  printf("\t,0,0,0,0\n");
+  if (CDecl->getNumIntfRefProtocols() > 0)
+    printf("\t,0,(struct _objc_protocol_list*)&_OBJC_CLASS_PROTOCOLS_%s,0,0\n", 
+           CDecl->getName());
+  else
+    printf("\t,0,0,0,0\n");
   printf("};\n");
   
   // class metadata generation.
@@ -724,6 +728,10 @@
   if (ClsDefCount == 0 && CatDefCount == 0)
     return;
   
+  // TODO: This is temporary until we decide how to access objc types in a
+  // c program
+  printf("\n#include <Objc/objc.h>\n");
+  
   // For each implemented class, write out all its meta data.
   for (int i = 0; i < ClsDefCount; i++)
     RewriteObjcClassMetaData(ClassImplementation[i]);