diff --git a/MLeaksFinder/UIViewController+MemoryLeak.m b/MLeaksFinder/UIViewController+MemoryLeak.m
index 5c59bf6..6a57797 100644
--- a/MLeaksFinder/UIViewController+MemoryLeak.m
+++ b/MLeaksFinder/UIViewController+MemoryLeak.m
@@ -53,7 +53,7 @@ - (void)swizzled_dismissViewControllerAnimated:(BOOL)flag completion:(void (^)(v
     
     if (!dismissedViewController) return;
     
-    [dismissedViewController willDealloc];
+    objc_setAssociatedObject(self, kHasBeenPoppedKey, @(YES), OBJC_ASSOCIATION_RETAIN);
 }
 
 - (BOOL)willDealloc {