diff --git a/src/frontend/src/components/codeAreaComponent/index.tsx b/src/frontend/src/components/codeAreaComponent/index.tsx
index 9772b3199..00721137f 100644
--- a/src/frontend/src/components/codeAreaComponent/index.tsx
+++ b/src/frontend/src/components/codeAreaComponent/index.tsx
@@ -46,7 +46,7 @@ export default function CodeAreaComponent({
           onChange(value);
         }}
       >
-        <div className="flex w-full items-center">
+        <div className="flex w-full items-center gap-3">
           <span
             id={id}
             data-testid={id}
@@ -63,7 +63,7 @@ export default function CodeAreaComponent({
             <IconComponent
               name="ExternalLink"
               className={
-                "icons-parameters-comp" +
+                "icons-parameters-comp shrink-0" +
                 (disabled ? " text-ring" : " hover:text-accent-foreground")
               }
             />
diff --git a/src/frontend/src/components/inputFileComponent/index.tsx b/src/frontend/src/components/inputFileComponent/index.tsx
index 8eee0ec4e..9710f7f61 100644
--- a/src/frontend/src/components/inputFileComponent/index.tsx
+++ b/src/frontend/src/components/inputFileComponent/index.tsx
@@ -99,7 +99,7 @@ export default function InputFileComponent({
 
   return (
     <div className={disabled ? "input-component-div" : "w-full"}>
-      <div className="input-file-component">
+      <div className="input-file-component gap-3">
         <span
           onClick={handleButtonClick}
           className={
@@ -115,14 +115,14 @@ export default function InputFileComponent({
         {!editNode && (
           <Button
             unstyled
-            className="ml-3 inline-flex items-center justify-center"
+            className="inline-flex items-center justify-center"
             onClick={handleButtonClick}
             loading={loading}
             disabled={disabled}
           >
             <IconComponent
               name="FileSearch2"
-              className={"icons-parameters-comp"}
+              className="icons-parameters-comp shrink-0"
             />
           </Button>
         )}
diff --git a/src/frontend/src/components/promptComponent/index.tsx b/src/frontend/src/components/promptComponent/index.tsx
index 99e44ba5d..a4c1f9225 100644
--- a/src/frontend/src/components/promptComponent/index.tsx
+++ b/src/frontend/src/components/promptComponent/index.tsx
@@ -36,7 +36,7 @@ export default function PromptAreaComponent({
         nodeClass={nodeClass}
         setNodeClass={setNodeClass}
       >
-        <div className="flex w-full items-center">
+        <div className="flex w-full items-center gap-3">
           <span
             id={id}
             data-testid={id}
@@ -54,7 +54,7 @@ export default function PromptAreaComponent({
               id={id}
               name="ExternalLink"
               className={
-                "icons-parameters-comp" +
+                "icons-parameters-comp shrink-0" +
                 (disabled ? " text-ring" : " hover:text-accent-foreground")
               }
             />
diff --git a/src/frontend/src/components/textAreaComponent/index.tsx b/src/frontend/src/components/textAreaComponent/index.tsx
index 7bd10153b..b49574920 100644
--- a/src/frontend/src/components/textAreaComponent/index.tsx
+++ b/src/frontend/src/components/textAreaComponent/index.tsx
@@ -22,7 +22,7 @@ export default function TextAreaComponent({
 
   return (
     <div className={"flex w-full items-center " + (disabled ? "" : "")}>
-      <div className="flex w-full items-center" data-testid={"div-" + id}>
+      <div className="flex w-full items-center gap-3" data-testid={"div-" + id}>
         <Input
           id={id}
           data-testid={id}
@@ -34,7 +34,7 @@ export default function TextAreaComponent({
             onChange(event.target.value);
           }}
         />
-        <div>
+        <div className="flex items-center">
           <GenericModal
             type={TypeModal.TEXT}
             buttonText="Finish Editing"
@@ -50,7 +50,7 @@ export default function TextAreaComponent({
                 id={id}
                 name="ExternalLink"
                 className={
-                  "icons-parameters-comp w-[1.35rem]" +
+                  "icons-parameters-comp shrink-0" +
                   (disabled ? " text-ring" : " hover:text-accent-foreground")
                 }
               />