From b4b41801ca339edeff6a908c850079ea0e1372c7 Mon Sep 17 00:00:00 2001
From: Suphon Thanakornpakapong <paphonb@gmail.com>
Date: Sun, 17 Oct 2021 22:37:38 +0700
Subject: [PATCH] Fix work mode switch bugging out when entering search

---
 src/com/android/launcher3/allapps/AllAppsContainerView.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/com/android/launcher3/allapps/AllAppsContainerView.java b/src/com/android/launcher3/allapps/AllAppsContainerView.java
index 5a2d167f8c..a7c75ad826 100644
--- a/src/com/android/launcher3/allapps/AllAppsContainerView.java
+++ b/src/com/android/launcher3/allapps/AllAppsContainerView.java
@@ -481,6 +481,10 @@ public class AllAppsContainerView extends StretchRecyclerViewContainer implement
                     });
             onActivePageChanged(mViewPager.getNextPage());
         } else {
+            if (mWorkModeSwitch != null) {
+                removeView(mWorkModeSwitch);
+                mWorkModeSwitch = null;
+            }
             mAH[AdapterHolder.MAIN].setup(findViewById(R.id.apps_list_view), null);
             mAH[AdapterHolder.WORK].recyclerView = null;
         }