diff --git a/libs/WindowManager/Shell/src/com/android/wm/shell/splitscreen/StageCoordinator.java b/libs/WindowManager/Shell/src/com/android/wm/shell/splitscreen/StageCoordinator.java
index bb1d06fbbbf2e..66989329860b6 100644
--- a/libs/WindowManager/Shell/src/com/android/wm/shell/splitscreen/StageCoordinator.java
+++ b/libs/WindowManager/Shell/src/com/android/wm/shell/splitscreen/StageCoordinator.java
@@ -609,7 +609,7 @@ class StageCoordinator implements SplitLayout.SplitLayoutHandler,
 
     private void onFoldedStateChanged(boolean folded) {
         if (folded && mMainStage.isActive()) {
-            exitSplitScreen();
+            exitSplitScreen(mMainStage);
         }
     }