diff --git a/src/frontend/src/pages/MainPage/components/componentsComponent/index.tsx b/src/frontend/src/pages/MainPage/components/componentsComponent/index.tsx
index 62ec3b246..fa5524891 100644
--- a/src/frontend/src/pages/MainPage/components/componentsComponent/index.tsx
+++ b/src/frontend/src/pages/MainPage/components/componentsComponent/index.tsx
@@ -270,18 +270,6 @@ export default function ComponentsComponent({
   return (
     <>
       <div className="flex w-full gap-4 pb-5">
-        <Button
-          variant="outline"
-          size="sm"
-          onClick={handleImport}
-          className="flex-shrink-0 text-sm"
-        >
-          <ForwardedIconComponent
-            name="FileUp"
-            className="h-5 w-5 text-primary"
-          />
-          Upload Flow
-        </Button>
         {allFlows?.length > 0 && (
           <HeaderComponent
             handleDelete={() => handleSelectOptionsChange("delete")}
diff --git a/src/frontend/src/pages/MainPage/components/headerComponent/index.tsx b/src/frontend/src/pages/MainPage/components/headerComponent/index.tsx
index ea45bb5aa..527c8d423 100644
--- a/src/frontend/src/pages/MainPage/components/headerComponent/index.tsx
+++ b/src/frontend/src/pages/MainPage/components/headerComponent/index.tsx
@@ -31,7 +31,7 @@ const HeaderComponent = ({
 
   return (
     <>
-      <div className="grid w-full grid-cols-3">
+      <div className="flex w-full items-center justify-between gap-4">
         <div className="flex items-center justify-self-start">
           <Button
             variant="none"
@@ -47,7 +47,7 @@ const HeaderComponent = ({
             </div>
           </Button>
         </div>
-        <div className="col-span-2 flex grid-cols-1 items-center gap-2 justify-self-end">
+        <div className="flex items-center gap-2">
           <div>
             <ShadTooltip
               content={
diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts
index 32de7dee2..329ca1fa0 100644
--- a/src/frontend/src/utils/styleUtils.ts
+++ b/src/frontend/src/utils/styleUtils.ts
@@ -56,6 +56,7 @@ import {
   FolderIcon,
   FolderPlus,
   FolderPlusIcon,
+  FolderUp,
   FormInput,
   Forward,
   Gift,
@@ -431,6 +432,7 @@ export const nodeIconsLucide: iconsType = {
   ChevronLeft,
   SlidersHorizontal,
   Palette,
+  FolderUp,
   Blocks,
   ChevronDown,
   ArrowLeft,