diff --git a/src/frontend/src/components/AccordionComponent/index.tsx b/src/frontend/src/components/AccordionComponent/index.tsx
index e00932c76..5c6ca0089 100644
--- a/src/frontend/src/components/AccordionComponent/index.tsx
+++ b/src/frontend/src/components/AccordionComponent/index.tsx
@@ -49,7 +49,7 @@ export default function AccordionComponent({
           >
             {trigger}
           </AccordionTrigger>
-          <AccordionContent className="AccordionContent">
+          <AccordionContent className="AccordionContent flex flex-col">
             {children}
           </AccordionContent>
         </AccordionItem>
diff --git a/src/frontend/src/components/IOview/index.tsx b/src/frontend/src/components/IOview/index.tsx
index 3b1059fe4..4ac7bb00d 100644
--- a/src/frontend/src/components/IOview/index.tsx
+++ b/src/frontend/src/components/IOview/index.tsx
@@ -265,28 +265,25 @@ export default function IOView({ children, open, setOpen }): JSX.Element {
                       </button>
                       {selectedViewField.type}
                     </div>
-                    <div className="flex h-full w-full">
-                      {inputs.some(
-                        (input) => input.id === selectedViewField.id
-                      ) ? (
-                        <IOInputField
-                          inputType={selectedViewField.type!}
-                          inputId={selectedViewField.id!}
-                        />
-                      ) : (
-                        <IOOutputView
-                          outputType={selectedViewField.type!}
-                          outputId={selectedViewField.id!}
-                        />
-                      )}
+                    <div className="h-full">
+                    {inputs.some(
+                      (input) => input.id === selectedViewField.id
+                    ) ? (
+                      <IOInputField
+                        inputType={selectedViewField.type!}
+                        inputId={selectedViewField.id!}
+                      />
+                    ) : (
+                      <IOOutputView
+                        outputType={selectedViewField.type!}
+                        outputId={selectedViewField.id!}
+                      />
+                    )}
                     </div>
                   </div>
                 )}
                 <div
-                  className={cn(
-                    "flex h-full w-full",
-                    selectedViewField ? "hidden" : ""
-                  )}
+                  className={cn("flex w-full h-full",selectedViewField ? "hidden" : "")}
                 >
                   <NewChatView
                     sendMessage={sendMessage}
diff --git a/src/frontend/src/modals/baseModal/index.tsx b/src/frontend/src/modals/baseModal/index.tsx
index 544fe9b70..6788fdcf9 100644
--- a/src/frontend/src/modals/baseModal/index.tsx
+++ b/src/frontend/src/modals/baseModal/index.tsx
@@ -21,7 +21,7 @@ type TriggerProps = {
 };
 
 const Content: React.FC<ContentProps> = ({ children }) => {
-  return <div className="h-full w-full">{children}</div>;
+  return <div className="h-full w-full flex flex-col">{children}</div>;
 };
 const Trigger: React.FC<TriggerProps> = ({ children, asChild, disable }) => {
   return (