From ba082ccc2f9bc334e5fced359209f56aa8fd78f1 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Wed, 11 Jun 2025 11:45:19 +1000
Subject: [PATCH] fix(ui): wait until last queue item deleted before flagging
 canvas session finished

---
 .../StagingArea/StagingAreaToolbarDiscardSelectedButton.tsx   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/invokeai/frontend/web/src/features/controlLayers/components/StagingArea/StagingAreaToolbarDiscardSelectedButton.tsx b/invokeai/frontend/web/src/features/controlLayers/components/StagingArea/StagingAreaToolbarDiscardSelectedButton.tsx
index b6fc9d934a..5554ce9744 100644
--- a/invokeai/frontend/web/src/features/controlLayers/components/StagingArea/StagingAreaToolbarDiscardSelectedButton.tsx
+++ b/invokeai/frontend/web/src/features/controlLayers/components/StagingArea/StagingAreaToolbarDiscardSelectedButton.tsx
@@ -16,15 +16,15 @@ export const StagingAreaToolbarDiscardSelectedButton = memo(({ isDisabled }: { i
 
   const { t } = useTranslation();
 
-  const discardSelected = useCallback(() => {
+  const discardSelected = useCallback(async () => {
     if (selectedItemId === null) {
       return;
     }
+    await deleteQueueItem.trigger(selectedItemId);
     const itemCount = ctx.$itemCount.get();
     if (itemCount <= 1) {
       dispatch(canvasSessionGenerationFinished());
     }
-    deleteQueueItem.trigger(selectedItemId);
   }, [selectedItemId, ctx.$itemCount, deleteQueueItem, dispatch]);
 
   return (