Remove associated task views if task is removed while Overview is showing

Bug: 78660939
Change-Id: Iccfb09c985662c150f42d382b92d17f853abaaea
diff --git a/quickstep/src/com/android/quickstep/views/RecentsView.java b/quickstep/src/com/android/quickstep/views/RecentsView.java
index f521b25..8d6b890 100644
--- a/quickstep/src/com/android/quickstep/views/RecentsView.java
+++ b/quickstep/src/com/android/quickstep/views/RecentsView.java
@@ -144,6 +144,14 @@
             // TODO: Re-enable layout transitions for addition of the unpinned task
             reloadIfNeeded();
         }
+
+        @Override
+        public void onTaskRemoved(int taskId) {
+            TaskView taskView = getTaskView(taskId);
+            if (taskView != null) {
+                dismissTask(taskView, true /* animate */, false /* removeTask */);
+            }
+        }
     };
 
     private int mLoadPlanId = -1;