diff --git a/Example 1/Classes/Model/ObjCClass/ObjCClassWrapper.m b/Example 1/Classes/Model/ObjCClass/ObjCClassWrapper.m
index 637de10..d8fd7bd 100644
--- a/Example 1/Classes/Model/ObjCClass/ObjCClassWrapper.m	
+++ b/Example 1/Classes/Model/ObjCClass/ObjCClassWrapper.m	
@@ -115,7 +115,7 @@ - (NSArray *) subclasses
         Class* classes = NULL;
         while (numClasses < newNumClasses) {
             numClasses = newNumClasses;
-            classes = realloc(classes, sizeof(Class) * numClasses);
+            classes = (Class*)realloc(classes, sizeof(Class) * numClasses);
             newNumClasses = objc_getClassList(classes, numClasses);
         }
 
diff --git a/PSTreeGraphView/PSBaseTreeGraphView.m b/PSTreeGraphView/PSBaseTreeGraphView.m
index 2072e36..a86bcc6 100644
--- a/PSTreeGraphView/PSBaseTreeGraphView.m
+++ b/PSTreeGraphView/PSBaseTreeGraphView.m
@@ -32,8 +32,8 @@ @interface PSBaseTreeGraphView ()
 	// Model
     id <PSTreeGraphModelNode> modelRoot_;
     
-	// Delegate
-	id <PSTreeGraphDelegate> delegate_;
+    // Delegate
+    __weak id <PSTreeGraphDelegate> delegate_;
     
     // Model Object -> SubtreeView Mapping
 	NSMutableDictionary *modelNodeToSubtreeViewMapTable_;