fix

Author: felixfeng33

apps/www/src/registry/default/plate-ui/equation-popover.tsx

@@ -13,7 +13,6 @@ import {
   useEditorRef,
   useElement,
   useReadOnly,
-  useSelected,
 } from '@udecode/plate/react';
 import { useEquationInput } from '@udecode/plate-math/react';
 import { BlockSelectionPlugin } from '@udecode/plate-selection/react';
@@ -40,13 +39,12 @@ const EquationPopoverContent = ({
   const editor = useEditorRef();
   const readOnly = useReadOnly();
   const element = useElement<TEquationElement>();
-  const selected = useSelected();
 
   useEffect(() => {
-    if (isInline && selected) {
+    if (isInline && open) {
       setOpen(true);
     }
-  }, [selected, isInline, setOpen]);
+  }, [isInline, open, setOpen]);
 
   if (readOnly) return null;
 

apps/www/src/registry/default/plate-ui/inline-equation-element.tsx

@@ -5,7 +5,11 @@ import { useRef, useState } from 'react';
 import type { TEquationElement } from '@udecode/plate-math';
 
 import { cn, withRef } from '@udecode/cn';
-import { useElement, useSelected } from '@udecode/plate/react';
+import {
+  useEditorSelector,
+  useElement,
+  useSelected,
+} from '@udecode/plate/react';
 import { useEquationElement } from '@udecode/plate-math/react';
 import { RadicalIcon } from 'lucide-react';
 
@@ -18,7 +22,11 @@ export const InlineEquationElement = withRef<typeof PlateElement>(
     const element = useElement<TEquationElement>();
     const katexRef = useRef<HTMLDivElement | null>(null);
     const selected = useSelected();
-    const [open, setOpen] = useState(selected);
+    const isCollapsed = useEditorSelector(
+      (editor) => editor.api.isCollapsed(),
+      []
+    );
+    const [open, setOpen] = useState(selected && isCollapsed);
 
     useEquationElement({
       element,