diff --git a/packages/slate-react/src/components/slate.tsx b/packages/slate-react/src/components/slate.tsx
index 0dedd53592..0a03ed82b2 100644
--- a/packages/slate-react/src/components/slate.tsx
+++ b/packages/slate-react/src/components/slate.tsx
@@ -61,9 +61,9 @@ export const Slate = (props: {
     handleSelectorChange(editor)
   }, [onChange])
 
-  EDITOR_TO_ON_CHANGE.set(editor, onContextChange)
-
   useEffect(() => {
+    EDITOR_TO_ON_CHANGE.set(editor, onContextChange)
+
     return () => {
       EDITOR_TO_ON_CHANGE.set(editor, () => {})
       unmountRef.current = true