diff --git a/tests/FlickerTests/src/com/android/server/wm/flicker/ime/CloseImeEditorPopupDialogTest.kt b/tests/FlickerTests/src/com/android/server/wm/flicker/ime/CloseImeEditorPopupDialogTest.kt
index bff099e2e7a19..6257484be9bde 100644
--- a/tests/FlickerTests/src/com/android/server/wm/flicker/ime/CloseImeEditorPopupDialogTest.kt
+++ b/tests/FlickerTests/src/com/android/server/wm/flicker/ime/CloseImeEditorPopupDialogTest.kt
@@ -91,6 +91,7 @@ class CloseImeEditorPopupDialogTest(private val testSpec: FlickerTestParameter)
                     .then()
                     .isVisible(FlickerComponentName.IME_SNAPSHOT)
                     .then()
+                    .isInvisible(FlickerComponentName.IME_SNAPSHOT)
                     .isInvisible(FlickerComponentName.IME)
         }
     }