diff --git a/src/components/chat-components/ChatControls.tsx b/src/components/chat-components/ChatControls.tsx
index 93c786a7..e900358d 100644
--- a/src/components/chat-components/ChatControls.tsx
+++ b/src/components/chat-components/ChatControls.tsx
@@ -142,7 +142,7 @@ const ChatControls: React.FC<ChatControlsProps> = ({
                 <ChevronDown size={10} />
               </DropdownMenu.Trigger>
 
-              <DropdownMenu.Portal>
+              <DropdownMenu.Portal container={activeDocument.body}>
                 <DropdownMenu.Content className="chain-select-content" align="end" sideOffset={5}>
                   <DropdownMenu.Item
                     onSelect={() => handleChainChange({ value: ChainType.LLM_CHAIN })}
diff --git a/src/components/chat-components/ChatInput.tsx b/src/components/chat-components/ChatInput.tsx
index 791ffa2a..bcc1fd6e 100644
--- a/src/components/chat-components/ChatInput.tsx
+++ b/src/components/chat-components/ChatInput.tsx
@@ -452,7 +452,7 @@ const ChatInput = forwardRef<{ focus: () => void }, ChatInputProps>(
                 <ChevronUp size={10} />
               </DropdownMenu.Trigger>
 
-              <DropdownMenu.Portal>
+              <DropdownMenu.Portal container={activeDocument.body}>
                 <DropdownMenu.Content className="model-select-content" align="start">
                   {settings.activeModels
                     .filter((model) => model.enabled)
diff --git a/src/components/chat-components/TooltipActionButton.tsx b/src/components/chat-components/TooltipActionButton.tsx
index 874c28b8..0f4e3bca 100644
--- a/src/components/chat-components/TooltipActionButton.tsx
+++ b/src/components/chat-components/TooltipActionButton.tsx
@@ -15,7 +15,7 @@ export function TooltipActionButton({ onClick, Icon, children }: PropsWithChildr
           {Icon}
         </button>
       </Tooltip.Trigger>
-      <Tooltip.Portal>
+      <Tooltip.Portal container={activeDocument.body}>
         <Tooltip.Content sideOffset={5} className="tooltip-text">
           {children}
         </Tooltip.Content>