diff --git a/go/quickstep/src/com/android/quickstep/TaskOverlayFactoryGo.java b/go/quickstep/src/com/android/quickstep/TaskOverlayFactoryGo.java
index f3c7a02..fc07162 100644
--- a/go/quickstep/src/com/android/quickstep/TaskOverlayFactoryGo.java
+++ b/go/quickstep/src/com/android/quickstep/TaskOverlayFactoryGo.java
@@ -243,7 +243,7 @@
             @SuppressLint("NewApi")
             public void onListen() {
                 if (mIsAllowedByPolicy) {
-                    sendNIUIntent(ACTION_LISTEN);
+                    endLiveTileMode(() -> sendNIUIntent(ACTION_LISTEN));
                 } else {
                     showBlockedByPolicyMessage();
                 }
@@ -252,7 +252,7 @@
             @SuppressLint("NewApi")
             public void onTranslate() {
                 if (mIsAllowedByPolicy) {
-                    sendNIUIntent(ACTION_TRANSLATE);
+                    endLiveTileMode(() -> sendNIUIntent(ACTION_TRANSLATE));
                 } else {
                     showBlockedByPolicyMessage();
                 }
@@ -261,7 +261,7 @@
             @SuppressLint("NewApi")
             public void onSearch() {
                 if (mIsAllowedByPolicy) {
-                    sendNIUIntent(ACTION_SEARCH);
+                    endLiveTileMode(() -> sendNIUIntent(ACTION_SEARCH));
                 } else {
                     showBlockedByPolicyMessage();
                 }