From 7bf8d7cd5372cbd396ca9ee2da0a10a72649361e Mon Sep 17 00:00:00 2001
From: Ashley Wulber <ashley@system76.com>
Date: Thu, 20 Nov 2025 13:32:52 -0500
Subject: [PATCH] fix: better surface view cleanup

---
 src/dialog.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/dialog.rs b/src/dialog.rs
index 843ce94..270f026 100644
--- a/src/dialog.rs
+++ b/src/dialog.rs
@@ -373,7 +373,7 @@ impl<M: Send + 'static> Dialog<M> {
             if !self.cosmic.surface_views.is_empty() {
                 log::debug!("waiting for surfaces to close...");
                 let mut tasks = Vec::new();
-                for id in self.cosmic.surface_views.drain() {
+                for id in self.cosmic.surface_views.iter() {
                     match id.1.1 {
                         SurfaceIdWrapper::Window(id) => {
                             tasks.push(window::close::<M>(id).discard());
@@ -403,6 +403,7 @@ impl<M: Send + 'static> Dialog<M> {
                 tasks.push(Task::future(async move {
                     cosmic::action::app(on_result_message)
                 }));
+                tasks.push(command);
                 return Task::batch(tasks);
             }
             let on_result_message = (self.on_result)(result);