diff --git a/lib/ProMotion/screens/_screen_module.rb b/lib/ProMotion/screens/_screen_module.rb
index f933244..1aef2ec 100644
--- a/lib/ProMotion/screens/_screen_module.rb
+++ b/lib/ProMotion/screens/_screen_module.rb
@@ -140,10 +140,6 @@ module ProMotion
     end
     def on_disappear; end
 
-    def main_controller
-      self.navigation_controller || self
-    end
-
     def view_controller
       warn "[DEPRECATION] `view_controller` is deprecated, as screens are now UIViewController subclasses."
       self
diff --git a/lib/ProMotion/screens/behaves_like_screen.rb b/lib/ProMotion/screens/behaves_like_screen.rb
index e84538a..eaf1f17 100644
--- a/lib/ProMotion/screens/behaves_like_screen.rb
+++ b/lib/ProMotion/screens/behaves_like_screen.rb
@@ -1,9 +1,10 @@
 module ProMotion
-	module BehavesLikeScreen
-		def pm_main_controller
-			respond_to?(:main_controller) ? main_controller : self
-		end
-	end
+  module BehavesLikeScreen
+    def pm_main_controller
+      navigationController || self
+    end
+    alias_method :main_controller, :pm_main_controller
+  end
 end
 
 UIViewController.send :include, ProMotion::BehavesLikeScreen