diff --git a/frontend/src/components/ai-elements/model-selector.tsx b/frontend/src/components/ai-elements/model-selector.tsx
index ef6ebd7..dc8682b 100644
--- a/frontend/src/components/ai-elements/model-selector.tsx
+++ b/frontend/src/components/ai-elements/model-selector.tsx
@@ -188,8 +188,8 @@ export const ModelSelectorLogoGroup = ({
 }: ModelSelectorLogoGroupProps) => (
   <div
     className={cn(
-      "-space-x-1 flex shrink-0 items-center [&>img]:rounded-full [&>img]:bg-background [&>img]:p-px [&>img]:ring-1 dark:[&>img]:bg-foreground",
-      className
+      "[&>img]:bg-background dark:[&>img]:bg-foreground flex shrink-0 items-center -space-x-1 [&>img]:rounded-full [&>img]:p-px [&>img]:ring-1",
+      className,
     )}
     {...props}
   />
@@ -201,5 +201,8 @@ export const ModelSelectorName = ({
   className,
   ...props
 }: ModelSelectorNameProps) => (
-  <span className={cn("flex-1 truncate text-left", className)} {...props} />
+  <span
+    className={cn("flex-1 truncate text-left text-xs", className)}
+    {...props}
+  />
 );
diff --git a/frontend/src/components/ai-elements/prompt-input.tsx b/frontend/src/components/ai-elements/prompt-input.tsx
index bd2fee8..d18b3a6 100644
--- a/frontend/src/components/ai-elements/prompt-input.tsx
+++ b/frontend/src/components/ai-elements/prompt-input.tsx
@@ -962,13 +962,10 @@ export const PromptInputButton = ({
   size,
   ...props
 }: PromptInputButtonProps) => {
-  const newSize =
-    size ?? (Children.count(props.children) > 1 ? "sm" : "icon-sm");
-
   return (
     <InputGroupButton
       className={cn(className)}
-      size={newSize}
+      size="sm"
       type="button"
       variant={variant}
       {...props}
diff --git a/frontend/src/components/workspace/input-box.tsx b/frontend/src/components/workspace/input-box.tsx
index e7ccd9a..057fbc1 100644
--- a/frontend/src/components/workspace/input-box.tsx
+++ b/frontend/src/components/workspace/input-box.tsx
@@ -144,8 +144,7 @@ export function InputBox({
           >
             <ModelSelectorTrigger asChild>
               <PromptInputButton>
-                <BoxIcon className="size-4" />
-                <ModelSelectorName className="font-light">
+                <ModelSelectorName className="text-xs font-normal">
                   {selectedModel?.displayName}
                 </ModelSelectorName>
               </PromptInputButton>