diff --git a/src/frontend/src/modals/IOModal/index.tsx b/src/frontend/src/modals/IOModal/index.tsx
index 5f6d455be..3e28552d5 100644
--- a/src/frontend/src/modals/IOModal/index.tsx
+++ b/src/frontend/src/modals/IOModal/index.tsx
@@ -139,7 +139,7 @@ export default function IOModal({
   useEffect(() => {
     setSelectedViewField(startView());
     if (haveChat) {
-      getMessagesTable("union").then(({ rows, columns }) => {
+      getMessagesTable("union", currentFlow!.id).then(({ rows, columns }) => {
         setMessages(rows);
         setColumns(columns);
       });