diff --git a/src/frontend/src/modals/formModal/index.tsx b/src/frontend/src/modals/formModal/index.tsx
index d74980183..3dd0e3ad9 100644
--- a/src/frontend/src/modals/formModal/index.tsx
+++ b/src/frontend/src/modals/formModal/index.tsx
@@ -214,7 +214,7 @@ export default function FormModal({
             intermediate_steps?: string;
             is_bot: boolean;
             message: string;
-            template: string;
+            prompt?: string;
             type: string;
             chatKey: string;
             files?: Array<any>;
@@ -225,7 +225,7 @@ export default function FormModal({
                   ? {
                       isSend: !chatItem.is_bot,
                       message: chatItem.message,
-                      template: chatItem.template,
+                      template: tabsState[flow.id].formKeysData.template,
                       thought: chatItem.intermediate_steps,
                       files: chatItem.files,
                       chatKey: chatItem.chatKey,
@@ -233,7 +233,7 @@ export default function FormModal({
                   : {
                       isSend: !chatItem.is_bot,
                       message: chatItem.message,
-                      template: chatItem.template,
+                      template: tabsState[flow.id].formKeysData.template,
                       thought: chatItem.intermediate_steps,
                       chatKey: chatItem.chatKey,
                     }