diff --git a/core/store/src/state/context/document.tsx b/core/store/src/state/context/document.tsx
index 057532f9e..ce3ce1092 100644
--- a/core/store/src/state/context/document.tsx
+++ b/core/store/src/state/context/document.tsx
@@ -33,9 +33,9 @@ export const useDocument = (docId: string) => {
   return store.docs[docId];
 };
 
-export const useGetDocument = () => (docId: string) => {
+export const useGetDocument = () => {
   const store = useStore();
-  return store.docs[docId];
+  return (docId: string) => store.docs[docId];
 };
 
 /**