diff --git a/src/frontend/apps/impress/src/features/docs/doc-management/api/useDuplicateDoc.tsx b/src/frontend/apps/impress/src/features/docs/doc-management/api/useDuplicateDoc.tsx
index e22733a0..de128ef1 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-management/api/useDuplicateDoc.tsx
+++ b/src/frontend/apps/impress/src/features/docs/doc-management/api/useDuplicateDoc.tsx
@@ -80,7 +80,7 @@ export function useDuplicateDoc(options?: DuplicateDocOptions) {
 
       return await duplicateDoc(variables);
     },
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       void queryClient.resetQueries({
         queryKey: [KEY_LIST_DOC],
       });
@@ -89,14 +89,14 @@ export function useDuplicateDoc(options?: DuplicateDocOptions) {
         duration: 3000,
       });
 
-      void options?.onSuccess?.(data, variables, context);
+      void options?.onSuccess?.(data, variables, onMutateResult, context);
     },
-    onError: (error, variables, context) => {
+    onError: (error, variables, onMutateResult, context) => {
       toast(t('Failed to duplicate the document...'), VariantType.ERROR, {
         duration: 3000,
       });
 
-      void options?.onError?.(error, variables, context);
+      void options?.onError?.(error, variables, onMutateResult, context);
     },
   });
 }
diff --git a/src/frontend/apps/impress/src/features/docs/doc-management/api/useRemoveDoc.tsx b/src/frontend/apps/impress/src/features/docs/doc-management/api/useRemoveDoc.tsx
index 32ad2301..9b70c8b6 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-management/api/useRemoveDoc.tsx
+++ b/src/frontend/apps/impress/src/features/docs/doc-management/api/useRemoveDoc.tsx
@@ -33,19 +33,19 @@ export const useRemoveDoc = ({
   return useMutation<void, APIError, RemoveDocProps>({
     mutationFn: removeDoc,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       listInvalidQueries?.forEach((queryKey) => {
         void queryClient.invalidateQueries({
           queryKey: [queryKey],
         });
       });
       if (options?.onSuccess) {
-        void options.onSuccess(data, variables, context);
+        void options.onSuccess(data, variables, onMutateResult, context);
       }
     },
-    onError: (error, variables, context) => {
+    onError: (error, variables, onMutateResult, context) => {
       if (options?.onError) {
-        void options.onError(error, variables, context);
+        void options.onError(error, variables, onMutateResult, context);
       }
     },
   });
diff --git a/src/frontend/apps/impress/src/features/docs/doc-management/api/useRestoreDoc.tsx b/src/frontend/apps/impress/src/features/docs/doc-management/api/useRestoreDoc.tsx
index ec7a47ff..8f8e5d79 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-management/api/useRestoreDoc.tsx
+++ b/src/frontend/apps/impress/src/features/docs/doc-management/api/useRestoreDoc.tsx
@@ -36,19 +36,19 @@ export const useRestoreDoc = ({
   return useMutation<void, APIError, RestoreDocProps>({
     mutationFn: restoreDoc,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       listInvalidQueries?.forEach((queryKey) => {
         void queryClient.invalidateQueries({
           queryKey: [queryKey],
         });
       });
       if (options?.onSuccess) {
-        void options.onSuccess(data, variables, context);
+        void options.onSuccess(data, variables, onMutateResult, context);
       }
     },
-    onError: (error, variables, context) => {
+    onError: (error, variables, onMutateResult, context) => {
       if (options?.onError) {
-        void options.onError(error, variables, context);
+        void options.onError(error, variables, onMutateResult, context);
       }
     },
   });
diff --git a/src/frontend/apps/impress/src/features/docs/doc-management/api/useUpdateDoc.tsx b/src/frontend/apps/impress/src/features/docs/doc-management/api/useUpdateDoc.tsx
index 0e5af2db..cdd8ad23 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-management/api/useUpdateDoc.tsx
+++ b/src/frontend/apps/impress/src/features/docs/doc-management/api/useUpdateDoc.tsx
@@ -42,7 +42,7 @@ export function useUpdateDoc(queryConfig?: UseUpdateDoc) {
   return useMutation<Doc, APIError, UpdateDocParams>({
     mutationFn: updateDoc,
     ...queryConfig,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       queryConfig?.listInvalideQueries?.forEach((queryKey) => {
         void queryClient.invalidateQueries({
           queryKey: [queryKey],
@@ -50,10 +50,10 @@ export function useUpdateDoc(queryConfig?: UseUpdateDoc) {
       });
 
       if (queryConfig?.onSuccess) {
-        void queryConfig.onSuccess(data, variables, context);
+        void queryConfig.onSuccess(data, variables, onMutateResult, context);
       }
     },
-    onError: (error, variables, context) => {
+    onError: (error, variables, onMutateResult, context) => {
       // If error it means the user is probably not allowed to edit the doc
       // so we invalidate the canEdit query to update the UI accordingly
       void queryClient.invalidateQueries({
@@ -61,7 +61,7 @@ export function useUpdateDoc(queryConfig?: UseUpdateDoc) {
       });
 
       if (queryConfig?.onError) {
-        queryConfig.onError(error, variables, context);
+        queryConfig.onError(error, variables, onMutateResult, context);
       }
     },
   });
diff --git a/src/frontend/apps/impress/src/features/docs/doc-share/api/useDeleteDocAccess.ts b/src/frontend/apps/impress/src/features/docs/doc-share/api/useDeleteDocAccess.ts
index 6b99e79e..7ddbf9f9 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-share/api/useDeleteDocAccess.ts
+++ b/src/frontend/apps/impress/src/features/docs/doc-share/api/useDeleteDocAccess.ts
@@ -45,7 +45,7 @@ export const useDeleteDocAccess = (options?: UseDeleteDocAccessOptions) => {
   return useMutation<void, APIError, DeleteDocAccessProps>({
     mutationFn: deleteDocAccess,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       void queryClient.invalidateQueries({
         queryKey: [KEY_LIST_DOC_ACCESSES],
       });
@@ -63,7 +63,7 @@ export const useDeleteDocAccess = (options?: UseDeleteDocAccessOptions) => {
         queryKey: [KEY_LIST_USER],
       });
       if (options?.onSuccess) {
-        void options.onSuccess(data, variables, context);
+        void options.onSuccess(data, variables, onMutateResult, context);
       }
     },
   });
diff --git a/src/frontend/apps/impress/src/features/docs/doc-share/api/useDeleteDocInvitation.ts b/src/frontend/apps/impress/src/features/docs/doc-share/api/useDeleteDocInvitation.ts
index 4805cfa1..3ba722ae 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-share/api/useDeleteDocInvitation.ts
+++ b/src/frontend/apps/impress/src/features/docs/doc-share/api/useDeleteDocInvitation.ts
@@ -53,17 +53,17 @@ export const useDeleteDocInvitation = (
   >({
     mutationFn: deleteDocInvitation,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       void queryClient.invalidateQueries({
         queryKey: [KEY_LIST_DOC_INVITATIONS],
       });
       if (options?.onSuccess) {
-        void options.onSuccess(data, variables, context);
+        void options.onSuccess(data, variables, onMutateResult, context);
       }
     },
-    onError: (error, variables, context) => {
+    onError: (error, variables, onMutateResult, context) => {
       if (options?.onError) {
-        void options.onError(error, variables, context);
+        void options.onError(error, variables, onMutateResult, context);
       }
     },
   });
diff --git a/src/frontend/apps/impress/src/features/docs/doc-share/api/useDocAccessRequest.tsx b/src/frontend/apps/impress/src/features/docs/doc-share/api/useDocAccessRequest.tsx
index 313d879d..9648a81f 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-share/api/useDocAccessRequest.tsx
+++ b/src/frontend/apps/impress/src/features/docs/doc-share/api/useDocAccessRequest.tsx
@@ -59,12 +59,12 @@ export function useCreateDocAccessRequest(
   return useMutation<void, APIError, CreateDocAccessRequestParams>({
     mutationFn: createDocAccessRequest,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       void queryClient.resetQueries({
         queryKey: [KEY_LIST_DOC_ACCESS_REQUESTS],
       });
 
-      void options?.onSuccess?.(data, variables, context);
+      void options?.onSuccess?.(data, variables, onMutateResult, context);
     },
   });
 }
@@ -169,7 +169,7 @@ export const useAcceptDocAccessRequest = (
   return useMutation<void, APIError, acceptDocAccessRequestsParams>({
     mutationFn: acceptDocAccessRequests,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       void queryClient.invalidateQueries({
         queryKey: [KEY_LIST_DOC_ACCESSES],
       });
@@ -179,7 +179,7 @@ export const useAcceptDocAccessRequest = (
       });
 
       if (options?.onSuccess) {
-        void options.onSuccess(data, variables, context);
+        void options.onSuccess(data, variables, onMutateResult, context);
       }
     },
   });
@@ -223,13 +223,13 @@ export const useDeleteDocAccessRequest = (
   return useMutation<void, APIError, DeleteDocAccessRequestParams>({
     mutationFn: deleteDocAccessRequest,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       void queryClient.invalidateQueries({
         queryKey: [KEY_LIST_DOC_ACCESS_REQUESTS],
       });
 
       if (options?.onSuccess) {
-        void options.onSuccess(data, variables, context);
+        void options.onSuccess(data, variables, onMutateResult, context);
       }
     },
   });
diff --git a/src/frontend/apps/impress/src/features/docs/doc-share/api/useUpdateDocAccess.ts b/src/frontend/apps/impress/src/features/docs/doc-share/api/useUpdateDocAccess.ts
index 58183dcb..05f24947 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-share/api/useUpdateDocAccess.ts
+++ b/src/frontend/apps/impress/src/features/docs/doc-share/api/useUpdateDocAccess.ts
@@ -50,7 +50,7 @@ export const useUpdateDocAccess = (options?: UseUpdateDocAccessOptions) => {
   return useMutation<Access, APIError, UpdateDocAccessProps>({
     mutationFn: updateDocAccess,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       void queryClient.invalidateQueries({
         queryKey: [KEY_LIST_DOC_ACCESSES],
       });
@@ -65,7 +65,7 @@ export const useUpdateDocAccess = (options?: UseUpdateDocAccessOptions) => {
         queryKey: [KEY_LIST_DOC],
       });
       if (options?.onSuccess) {
-        void options.onSuccess(data, variables, context);
+        void options.onSuccess(data, variables, onMutateResult, context);
       }
     },
   });
diff --git a/src/frontend/apps/impress/src/features/docs/doc-share/api/useUpdateDocInvitation.ts b/src/frontend/apps/impress/src/features/docs/doc-share/api/useUpdateDocInvitation.ts
index f7befde1..a7170ec0 100644
--- a/src/frontend/apps/impress/src/features/docs/doc-share/api/useUpdateDocInvitation.ts
+++ b/src/frontend/apps/impress/src/features/docs/doc-share/api/useUpdateDocInvitation.ts
@@ -62,17 +62,17 @@ export const useUpdateDocInvitation = (
   >({
     mutationFn: updateDocInvitation,
     ...options,
-    onSuccess: (data, variables, context) => {
+    onSuccess: (data, variables, onMutateResult, context) => {
       void queryClient.invalidateQueries({
         queryKey: [KEY_LIST_DOC_INVITATIONS],
       });
       if (options?.onSuccess) {
-        void options.onSuccess(data, variables, context);
+        void options.onSuccess(data, variables, onMutateResult, context);
       }
     },
-    onError: (error, variables, context) => {
+    onError: (error, variables, onMutateResult, context) => {
       if (options?.onError) {
-        void options.onError(error, variables, context);
+        void options.onError(error, variables, onMutateResult, context);
       }
     },
   });