diff --git a/webui2/src/components/code/file-viewer.tsx b/webui2/src/components/code/file-viewer.tsx
index ba51c001bc240d63889040ec900e9f6f51c9c874..b556c2b09b0a822929327125f17154868b220bfe 100644
--- a/webui2/src/components/code/file-viewer.tsx
+++ b/webui2/src/components/code/file-viewer.tsx
@@ -245,8 +245,7 @@ export function FileViewer({ blob }: FileViewerProps) {
         </span>
         <Button
           variant="ghost"
-          size="icon"
-          className="size-7"
+          size="icon-xs"
           onClick={copyToClipboard}
           title="Copy"
         >
diff --git a/webui2/src/components/layout/header.tsx b/webui2/src/components/layout/header.tsx
index e69f6ca8bf284ea3bdf7111a95da047dae95f3fa..5333ca4ae1f3bdb3bd35a6d1405d75bbe93061fc 100644
--- a/webui2/src/components/layout/header.tsx
+++ b/webui2/src/components/layout/header.tsx
@@ -59,7 +59,7 @@ export function Header() {
         <div className="ml-auto flex items-center gap-2">
           {mode === "readonly" && <span className="text-muted-foreground text-xs">Read only</span>}
 
-          <Button variant="ghost" size="icon" onClick={toggle} title="Toggle theme">
+          <Button variant="ghost" size="icon-sm" onClick={toggle} title="Toggle theme">
             {theme === "light" ? <Moon className="size-4" /> : <Sun className="size-4" />}
           </Button>