I suggest using jotai for atomic state management. What they're trying to do is nice, but I think it's full of unnecessary complexity.

https://jotai.org/