Merge "Keep the task in main stage at top after dismissed split due to fold" into sc-v2-dev am: 7b8f6bf5b2

Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/14928714

Change-Id: Ic8c5128144db028b74532c392ac9bfb5153f6311

libs/WindowManager/Shell/src/com/android/wm/shell/splitscreen/StageCoordinator.java

@@ -610,7 +610,7 @@ class StageCoordinator implements SplitLayout.SplitLayoutHandler,
 
     private void onFoldedStateChanged(boolean folded) {
         if (folded && mMainStage.isActive()) {
-            exitSplitScreen();
+            exitSplitScreen(mMainStage);
         }
     }