Address comments

sigmacomputing · Sep 18, 2024 · d50d609 · d50d609
1 parent cdecf54
commit d50d609
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/react/hooks.ts b/src/react/hooks.ts
@@ -202,10 +202,10 @@ export function useActionEffect(configId: string, effect: () => void) {
   const effectRef = useRef(effect);
 
   useEffect(() => {
-    const unsubscribe = client.config.registerEffect(
-      configId,
-      effectRef.current,
-    );
-    return unsubscribe;
+    effectRef.current = effect;
+  });
+
+  useEffect(() => {
+    return client.config.registerEffect(configId, effectRef.current);
   }, [client, configId, effect]);
 }