diff --git a/packages/xl-ai/src/components/AIMenu/AIMenuController.tsx b/packages/xl-ai/src/components/AIMenu/AIMenuController.tsx
index 6072d1b911..8a8b29d54a 100644
--- a/packages/xl-ai/src/components/AIMenu/AIMenuController.tsx
+++ b/packages/xl-ai/src/components/AIMenu/AIMenuController.tsx
@@ -5,7 +5,7 @@ import {
   useExtension,
   useExtensionState,
 } from "@blocknote/react";
-import { autoUpdate, offset, size } from "@floating-ui/react";
+import { autoUpdate, flip, offset, size } from "@floating-ui/react";
 import { FC, useMemo } from "react";
 
 import { AIExtension } from "../../AIExtension.js";
@@ -32,7 +32,7 @@ export const AIMenuController = (props: {
         placement: "bottom",
         middleware: [
           offset(10),
-          // flip(),
+          flip(),
           size({
             apply({ rects, elements }) {
               Object.assign(elements.floating.style, {