diff --git a/apps/viewer/src/features/chat/api/utils/executeGroup.ts b/apps/viewer/src/features/chat/api/utils/executeGroup.ts
index 475743cb42c..374f59d706e 100644
--- a/apps/viewer/src/features/chat/api/utils/executeGroup.ts
+++ b/apps/viewer/src/features/chat/api/utils/executeGroup.ts
@@ -23,7 +23,11 @@ import { computePaymentInputRuntimeOptions } from '@/features/blocks/inputs/paym
 import { injectVariableValuesInButtonsInputBlock } from '@/features/blocks/inputs/buttons/api/utils/injectVariableValuesInButtonsInputBlock'
 
 export const executeGroup =
-  (state: SessionState, currentReply?: ChatReply) =>
+  (
+    state: SessionState,
+    currentReply?: ChatReply,
+    currentLastBubbleId?: string
+  ) =>
   async (
     group: Group
   ): Promise<ChatReply & { newSessionState: SessionState }> => {
@@ -32,7 +36,7 @@ export const executeGroup =
       currentReply?.clientSideActions
     let logs: ChatReply['logs'] = currentReply?.logs
     let nextEdgeId = null
-    let lastBubbleBlockId: string | undefined
+    let lastBubbleBlockId: string | undefined = currentLastBubbleId
 
     let newSessionState = state
 
@@ -97,11 +101,15 @@ export const executeGroup =
       return { messages, newSessionState, clientSideActions, logs }
     }
 
-    return executeGroup(newSessionState, {
-      messages,
-      clientSideActions,
-      logs,
-    })(nextGroup.group)
+    return executeGroup(
+      newSessionState,
+      {
+        messages,
+        clientSideActions,
+        logs,
+      },
+      lastBubbleBlockId
+    )(nextGroup.group)
   }
 
 const computeRuntimeOptions =