diff --git a/__mocks__/zustand.ts b/__mocks__/zustand.ts
index 716939ec..954e1ed5 100644
--- a/__mocks__/zustand.ts
+++ b/__mocks__/zustand.ts
@@ -26,4 +26,4 @@ export const create = (f) => (f === undefined ? createImpl : createImpl(f));
 
 export default create;
 
-export { useStore } from 'zustand';
+export { useStoreWithEqualityFn as useStore } from 'zustand/traditional';