diff --git a/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/splitscreen/benchmark/SplitScreenBase.kt b/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/splitscreen/benchmark/SplitScreenBase.kt
index fdcb7e5a09843..a0e437c25aa78 100644
--- a/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/splitscreen/benchmark/SplitScreenBase.kt
+++ b/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/splitscreen/benchmark/SplitScreenBase.kt
@@ -33,8 +33,9 @@ abstract class SplitScreenBase(flicker: LegacyFlickerTest) : BaseBenchmarkTest(f
             tapl.setEnableRotation(true)
             setRotation(flicker.scenario.startRotation)
             tapl.setExpectedRotation(flicker.scenario.startRotation.value)
-            if (tapl.workspace.switchToOverview().hasTasks()) {
-                tapl.workspace.switchToOverview().dismissAllTasks()
+            val overview = tapl.workspace.switchToOverview()
+            if (overview.hasTasks()) {
+                overview.dismissAllTasks()
             }
         }
     }