diff --git a/src/reactivity.ts b/src/reactivity.ts
index 2a564be9..e507d048 100644
--- a/src/reactivity.ts
+++ b/src/reactivity.ts
@@ -233,9 +233,9 @@ export function useState<T extends object>(state: T): Reactive<T> {
       node,
       batched(() => node.render())
     );
+    onWillUnmount(() => clearReactivesForCallback(render));
   }
   const render = batchedRenderFunctions.get(node)!;
   const reactiveState = reactive(state, render);
-  onWillUnmount(() => clearReactivesForCallback(render));
   return reactiveState;
 }