Fix work mode switch bugging out when entering search

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;
         }