From bdab73701fb4ad3010be62a81a153e3e66e1032f Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Sat, 27 May 2023 18:31:01 +1000
Subject: [PATCH] fix(ui): canvas images not added to staging

---
 .../listeners/socketio/invocationComplete.ts             | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/invocationComplete.ts b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/invocationComplete.ts
index 95e6d831c0..60a3cdfedf 100644
--- a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/invocationComplete.ts
+++ b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/invocationComplete.ts
@@ -57,14 +57,7 @@ export const addInvocationCompleteListener = () => {
           graph_execution_state_id ===
           getState().canvas.layerState.stagingArea.sessionId
         ) {
-          const [{ payload: image }] = await take(
-            (
-              action
-            ): action is ReturnType<typeof imageMetadataReceived.fulfilled> =>
-              imageMetadataReceived.fulfilled.match(action) &&
-              action.payload.image_name === image_name
-          );
-          dispatch(addImageToStagingArea(image));
+          dispatch(addImageToStagingArea(imageDTO));
         }
 
         dispatch(progressImageSet(null));