diff --git a/src/frontend/src/components/IOview/index.tsx b/src/frontend/src/components/IOview/index.tsx
index 458bd7d1e..ae04bcba6 100644
--- a/src/frontend/src/components/IOview/index.tsx
+++ b/src/frontend/src/components/IOview/index.tsx
@@ -308,7 +308,9 @@ export default function IOView({ children, open, setOpen }): JSX.Element {
               <div className="absolute bottom-8 right-8"></div>
             )}
           </div>
-          {!haveChat && (
+        </div>
+      </BaseModal.Content>
+      <BaseModal.Footer>{!haveChat && (
             <div className="flex w-full justify-end pt-6">
               <Button
                 variant={"outline"}
@@ -327,9 +329,8 @@ export default function IOView({ children, open, setOpen }): JSX.Element {
                 Run Flow
               </Button>
             </div>
-          )}
-        </div>
-      </BaseModal.Content>
+          )}  
+      </BaseModal.Footer>
     </BaseModal>
   );
 }