Improve Jotai work (#18205)

packages/twenty-front/src/modules/front-components/hooks/useUnmountHeadlessFrontComponent.ts

@@ -1,21 +1,19 @@
 import { useCallback } from 'react';
 
 import { mountedHeadlessFrontComponentIdsState } from '@/front-components/states/mountedHeadlessFrontComponentIdsState';
-import { jotaiStore } from '@/ui/utilities/state/jotai/jotaiStore';
+import { useStore } from 'jotai';
 
 export const useUnmountHeadlessFrontComponent = () => {
+  const store = useStore();
   const unmountHeadlessFrontComponent = useCallback(
     (frontComponentId: string) => {
-      jotaiStore.set(
-        mountedHeadlessFrontComponentIdsState.atom,
-        (previousIds) => {
-          const next = new Set(previousIds);
-          next.delete(frontComponentId);
-          return next;
-        },
-      );
+      store.set(mountedHeadlessFrontComponentIdsState.atom, (previousIds) => {
+        const next = new Set(previousIds);
+        next.delete(frontComponentId);
+        return next;
+      });
     },
-    [],
+    [store],
   );
 
   return unmountHeadlessFrontComponent;