Fix crash

Zepo · May 12, 2016 · e346665 · e346665
1 parent 3456267
commit e346665
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/MLeaksFinder/UINavigationController+MemoryLeak.m b/MLeaksFinder/UINavigationController+MemoryLeak.m
@@ -50,8 +50,10 @@ - (UIViewController *)swizzled_popViewControllerAnimated:(BOOL)animated {
     }
 
     // VC is not dealloced until disappear when popped using a left-edge swipe gesture
-    extern const void *const kHasBeenPoppedKey;
-    objc_setAssociatedObject(poppedViewController, kHasBeenPoppedKey, @(YES), OBJC_ASSOCIATION_RETAIN);
+    if (poppedViewController) {
+        extern const void *const kHasBeenPoppedKey;
+        objc_setAssociatedObject(poppedViewController, kHasBeenPoppedKey, @(YES), OBJC_ASSOCIATION_RETAIN);
+    }
 
     return poppedViewController;
 }