diff --git a/src/features/compose/editor/plugins/focus-plugin.tsx b/src/features/compose/editor/plugins/focus-plugin.tsx index 65b567683..8a076f199 100644 --- a/src/features/compose/editor/plugins/focus-plugin.tsx +++ b/src/features/compose/editor/plugins/focus-plugin.tsx @@ -11,10 +11,6 @@ export const FOCUS_EDITOR_COMMAND: LexicalCommand = createCommand(); const FocusPlugin: React.FC = ({ autoFocus }) => { const [editor] = useLexicalComposerContext(); - const focus = () => { - editor.dispatchCommand(FOCUS_EDITOR_COMMAND, undefined); - }; - useEffect(() => editor.registerCommand(FOCUS_EDITOR_COMMAND, () => { editor.focus( () => { @@ -29,8 +25,10 @@ const FocusPlugin: React.FC = ({ autoFocus }) => { }, COMMAND_PRIORITY_NORMAL)); useEffect(() => { - if (autoFocus) focus(); - }, []); + if (autoFocus) { + editor.dispatchCommand(FOCUS_EDITOR_COMMAND, undefined); + } + }, [autoFocus, editor]); return null; };