diff --git a/frontend/src/app/socketio/listeners.ts b/frontend/src/app/socketio/listeners.ts
index c67fb512fa..d2766b6fdb 100644
--- a/frontend/src/app/socketio/listeners.ts
+++ b/frontend/src/app/socketio/listeners.ts
@@ -168,6 +168,7 @@ const makeSocketIOListeners = (
           setIntermediateImage({
             uuid: uuidv4(),
             ...data,
+            category: 'result',
           })
         );
         if (!data.isBase64) {