diff --git a/invokeai/frontend/web/src/services/events/onInvocationComplete.ts b/invokeai/frontend/web/src/services/events/onInvocationComplete.ts
index 6fcf808aab..03bc860d71 100644
--- a/invokeai/frontend/web/src/services/events/onInvocationComplete.ts
+++ b/invokeai/frontend/web/src/services/events/onInvocationComplete.ts
@@ -154,7 +154,7 @@ export const buildOnInvocationComplete = (
     // Update the node execution states - the image output is handled below
     if (data.origin === 'workflows') {
       await handleOriginWorkflows(data);
-    } else if (data.origin === 'canvas') {
+    } else if (data.origin === 'generation') {
       await handleOriginCanvas(data);
     } else {
       await handleOriginOther(data);
diff --git a/invokeai/frontend/web/src/services/events/setEventListeners.tsx b/invokeai/frontend/web/src/services/events/setEventListeners.tsx
index aff1200bc1..f0aaeb9ad6 100644
--- a/invokeai/frontend/web/src/services/events/setEventListeners.tsx
+++ b/invokeai/frontend/web/src/services/events/setEventListeners.tsx
@@ -131,7 +131,7 @@ export const setEventListeners = ({ socket, dispatch, getState, setIsConnected }
       }
     }
 
-    if (origin === 'canvas' && destination === 'canvas') {
+    if (origin === 'generation' && destination === 'canvas') {
       $lastCanvasProgressEvent.set(data);
     }
   });