docs: fix batching example

Omnistac · May 10, 2024 · bff2272 · bff2272
1 parent 19f944b
commit bff2272
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/docs/docs/advanced/batching.mdx b/docs/docs/advanced/batching.mdx
@@ -85,12 +85,15 @@ const manualBatchingAtom = atom('manualBatching', () => {
 
   // use `injectMemo` to "turn off" automatic batching:
   // highlight-next-line
-  const updateTwice = injectMemo(() => {
-    ecosystem.batch(() => {
-      store.setState(state => state + 1)
-      store.setState(state => state + 1)
-    })
-  }, [])
+  const updateTwice = injectMemo(
+    () => () => {
+      ecosystem.batch(() => {
+        store.setState(state => state + 1)
+        store.setState(state => state + 1)
+      })
+    },
+    []
+  )
 
   return api(store).setExports({ updateTwice })
 })