diff --git a/src/frontend/src/modals/formModal/index.tsx b/src/frontend/src/modals/formModal/index.tsx
index 9a786d776..07e63ac11 100644
--- a/src/frontend/src/modals/formModal/index.tsx
+++ b/src/frontend/src/modals/formModal/index.tsx
@@ -272,6 +272,7 @@ export default function FormModal({
     }
     if (data.type === "stream" && isStream) {
       if (data.prompt) {
+        updateLastMessage({ prompt: data.prompt });
       } else {
         updateLastMessage({ str: data.message });
       }