updated example

example/src/App.tsx

@@ -1,6 +1,6 @@
 import './App.css';
 import logo from './logo.svg';
-import { counterStore, useCounterStoreSelector } from './store/counterStore';
+import { counterStore, useCounterStoreSelector } from './stores/counterStore';
 
 function App() {
 	const count = useCounterStoreSelector('count');

example/src/stores/counterStore.ts

@@ -0,0 +1,14 @@
+import { createStore } from '@poly-state/poly-state';
+import { createStoreHooks } from '@poly-state/react';
+
+type CounterStore = {
+	count: number;
+};
+
+const counterStoreInitialState: CounterStore = {
+	count: 0,
+};
+
+export const counterStore = createStore(counterStoreInitialState);
+
+export const [useCounterStore, useCounterStoreSelector] = createStoreHooks(counterStore);

src/useStoreSelector.ts

@@ -10,19 +10,12 @@ export const useStoreSelector = <T extends StateConstraint, U extends keyof T>(
 	const subscriberRef = useRef<() => void>();
 
 	useEffect(() => {
-		//clean up previous listeners if dependencies change
-		if (subscriberRef.current) {
-			subscriberRef.current();
-		}
-
+		//clean up previous listener if dependencies change
+		subscriberRef.current?.();
 		subscriberRef.current = store.subscribeKey(key as keyof T, setState as any);
-
-		return () => {
-			if (subscriberRef?.current) {
-				subscriberRef.current();
-			}
-		};
 	}, [store, key]);
 
+	useEffect(() => () => subscriberRef.current?.());
+
 	return state;
 };