diff --git a/src/frontend/src/modals/IOModal/components/chatView/fileComponent/components/downloadButton/downloadButton.tsx b/src/frontend/src/modals/IOModal/components/chatView/fileComponent/components/downloadButton/downloadButton.tsx
index 8df64561b..4c1559e63 100644
--- a/src/frontend/src/modals/IOModal/components/chatView/fileComponent/components/downloadButton/downloadButton.tsx
+++ b/src/frontend/src/modals/IOModal/components/chatView/fileComponent/components/downloadButton/downloadButton.tsx
@@ -1,4 +1,5 @@
 import ForwardedIconComponent from "../../../../../../../components/genericIconComponent";
+import { Button } from "../../../../../../../components/ui/button";
 
 export default function DownloadButton({
   isHovered,
@@ -10,14 +11,18 @@ export default function DownloadButton({
   if (isHovered) {
     return (
       <div
-        className={`absolute right-1 top-1 rounded-bl-lg bg-muted text-sm font-bold text-foreground `}
+        className={`absolute right-1 top-1 rounded-md bg-muted text-sm font-bold text-foreground `}
       >
-        <button className="px-2 py-1 text-ring " onClick={handleDownload}>
+        <Button
+          variant={"none"}
+          className="bg-transparent px-2 py-1 text-ring"
+          onClick={handleDownload}
+        >
           <ForwardedIconComponent
             name="DownloadCloud"
-            className="h-5 w-5 text-current hover:scale-110"
+            className="h-5 w-5 bg-transparent text-current hover:scale-110"
           />
-        </button>
+        </Button>
       </div>
     );
   }