issue-1016: potential enhancement for traversing stacked controllers

CosmicMind · Jan 9, 2018 · cf88010 · cf88010
1 parent fb2a453
commit cf88010
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Sources/iOS/Material+UIViewController.swift b/Sources/iOS/Material+UIViewController.swift
@@ -42,7 +42,8 @@ internal extension UIViewController {
             if v is T {
                 return v as? T
             }
-            v = v?.parent as? TransitionController
+
+            v = v?.parent
         }
 
         return Application.rootViewController?.traverseTransitionViewControllerHierarchyForClassType()
@@ -55,6 +56,7 @@ internal extension UIViewController {
     func traverseTransitionViewControllerHierarchyForClassType<T: UIViewController>() -> T? {
         if let v = self as? T {
             return v
+
         } else if let v = self as? TransitionController {
             return v.rootViewController.traverseTransitionViewControllerHierarchyForClassType()
         }