|
@@ -34,7 +34,7 @@
|
34
|
34
|
_completionBlock = nil;
|
35
|
35
|
}
|
36
|
36
|
|
37
|
|
- [_presentedModals addObject:self.toVC.getLeafViewController];
|
|
37
|
+ [_presentedModals addObject:self.toVC.navigationController ? self.toVC.navigationController : self.toVC];
|
38
|
38
|
|
39
|
39
|
self.toVC = nil;
|
40
|
40
|
}];
|
|
@@ -119,7 +119,7 @@
|
119
|
119
|
}
|
120
|
120
|
|
121
|
121
|
- (void)dismissedModal:(UIViewController *)viewController {
|
122
|
|
- [_presentedModals removeObject:viewController];
|
|
122
|
+ [_presentedModals removeObject:viewController.navigationController ? viewController.navigationController : viewController];
|
123
|
123
|
[_delegate dismissedModal:viewController];
|
124
|
124
|
}
|
125
|
125
|
|