From 6c8270dae2e8663c9bde7f7b3b1f7a47c9291ef0 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Tue, 3 Oct 2023 13:04:19 +1100
Subject: [PATCH] fix(ui): canvas staging area works after undo

---
 .../frontend/web/src/features/canvas/store/canvasSelectors.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/invokeai/frontend/web/src/features/canvas/store/canvasSelectors.ts b/invokeai/frontend/web/src/features/canvas/store/canvasSelectors.ts
index 8f1e246aaa..509237944c 100644
--- a/invokeai/frontend/web/src/features/canvas/store/canvasSelectors.ts
+++ b/invokeai/frontend/web/src/features/canvas/store/canvasSelectors.ts
@@ -6,7 +6,9 @@ export const canvasSelector = (state: RootState): CanvasState => state.canvas;
 
 export const isStagingSelector = createSelector(
   [stateSelector],
-  ({ canvas }) => canvas.batchIds.length > 0
+  ({ canvas }) =>
+    canvas.batchIds.length > 0 ||
+    canvas.layerState.stagingArea.images.length > 0
 );
 
 export const initialCanvasImageSelector = (