spec/unit/split_screen_spec.rb in ProMotion-0.7.8 vs spec/unit/split_screen_spec.rb in ProMotion-1.0.0

@@ -26,15 +26,15 @@
     @app.window.rootViewController.should == @split_screen
   end
 
   it "should set the first viewController to MasterScreen" do
     @split_screen.master_screen.should == @master_screen
-    @split_screen.viewControllers.first.should == @master_screen.pm_main_controller
+    @split_screen.viewControllers.first.should == (@master_screen.navigationController || @master_screen)
   end
 
   it "should set the second viewController to DetailScreen" do
     @split_screen.detail_screen.should == @detail_screen
-    @split_screen.viewControllers.last.should == @detail_screen.pm_main_controller
+    @split_screen.viewControllers.last.should == (@detail_screen.navigationController || @detail_screen)
   end
 
   it "should set the title on both screens" do
     @master_screen.class.send(:get_title).should == "Master"
     @master_screen.title.should == "Master"