diff --git a/packages/embeds/js/package.json b/packages/embeds/js/package.json
index cc0d18900..dc747113d 100644
--- a/packages/embeds/js/package.json
+++ b/packages/embeds/js/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@typebot.io/js",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "description": "Javascript library to display typebots on your website",
   "type": "module",
   "main": "dist/index.js",
diff --git a/packages/embeds/js/src/components/ConversationContainer/ConversationContainer.tsx b/packages/embeds/js/src/components/ConversationContainer/ConversationContainer.tsx
index cf59647ef..57b6a03cf 100644
--- a/packages/embeds/js/src/components/ConversationContainer/ConversationContainer.tsx
+++ b/packages/embeds/js/src/components/ConversationContainer/ConversationContainer.tsx
@@ -171,7 +171,7 @@ export const ConversationContainer = (props: Props) => {
       setFormattedMessages([
         ...formattedMessages(),
         {
-          inputId: [...chatChunks()].pop()?.input?.id ?? '',
+          inputIndex: [...chatChunks()].length - 1,
           formattedMessage: data.lastMessageNewFormat as string,
         },
       ])
diff --git a/packages/embeds/js/src/components/InputChatBlock.tsx b/packages/embeds/js/src/components/InputChatBlock.tsx
index 0cbf82adf..e8f19597a 100644
--- a/packages/embeds/js/src/components/InputChatBlock.tsx
+++ b/packages/embeds/js/src/components/InputChatBlock.tsx
@@ -63,8 +63,8 @@ export const InputChatBlock = (props: Props) => {
   }
 
   createEffect(() => {
-    const formattedMessage = formattedMessages().find(
-      (message) => message.inputId === props.block.id
+    const formattedMessage = formattedMessages().findLast(
+      (message) => props.inputIndex === message.inputIndex
     )?.formattedMessage
     if (formattedMessage) setFormattedMessage(formattedMessage)
   })
diff --git a/packages/embeds/js/src/utils/formattedMessagesSignal.ts b/packages/embeds/js/src/utils/formattedMessagesSignal.ts
index 29ac208a7..265b57bd8 100644
--- a/packages/embeds/js/src/utils/formattedMessagesSignal.ts
+++ b/packages/embeds/js/src/utils/formattedMessagesSignal.ts
@@ -1,5 +1,5 @@
 import { createSignal } from 'solid-js'
 
 export const [formattedMessages, setFormattedMessages] = createSignal<
-  { inputId: string; formattedMessage: string }[]
+  { inputIndex: number; formattedMessage: string }[]
 >([])
diff --git a/packages/embeds/nextjs/package.json b/packages/embeds/nextjs/package.json
index ee2d6bcc6..f9d8427e4 100644
--- a/packages/embeds/nextjs/package.json
+++ b/packages/embeds/nextjs/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@typebot.io/nextjs",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "description": "Convenient library to display typebots on your Next.js website",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
diff --git a/packages/embeds/react/package.json b/packages/embeds/react/package.json
index 82d562b95..1347387e8 100644
--- a/packages/embeds/react/package.json
+++ b/packages/embeds/react/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@typebot.io/react",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "description": "Convenient library to display typebots on your React app",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",