diff --git a/packages/bruno-app/src/components/AIAssist/StyledWrapper.js b/packages/bruno-app/src/components/AIAssist/StyledWrapper.js
index 0205efb7b..e8210cece 100644
--- a/packages/bruno-app/src/components/AIAssist/StyledWrapper.js
+++ b/packages/bruno-app/src/components/AIAssist/StyledWrapper.js
@@ -5,6 +5,9 @@ const StyledWrapper = styled.div`
   top: 6px;
   right: 6px;
   z-index: 10;
+  background: ${(props) => props.theme.bg};
+  border: 1px solid ${(props) => props.theme.input.border};
+  border-radius: ${(props) => props.theme.border.radius.sm};
 
   .ai-assist-trigger {
     display: inline-flex;
@@ -13,11 +16,11 @@ const StyledWrapper = styled.div`
     width: 24px;
     height: 24px;
     border-radius: ${(props) => props.theme.border.radius.sm};
-    border: 1px solid transparent;
+    border: none;
     background: transparent;
     color: ${(props) => props.theme.colors.text.muted};
     cursor: pointer;
-    transition: color 0.15s ease, background-color 0.15s ease, border-color 0.15s ease;
+    transition: color 0.15s ease, background-color 0.15s ease;
     opacity: 0.7;
 
     &:hover,
@@ -25,7 +28,6 @@ const StyledWrapper = styled.div`
       opacity: 1;
       color: ${(props) => props.theme.colors.accent};
       background: ${(props) => props.theme.colors.accent}10;
-      border-color: ${(props) => props.theme.input.border};
     }
 
     &:focus-visible {