diff --git a/web/src/pages/memories/hooks.ts b/web/src/pages/memories/hooks.ts
index 483daa93c..dda708fd5 100644
--- a/web/src/pages/memories/hooks.ts
+++ b/web/src/pages/memories/hooks.ts
@@ -218,9 +218,6 @@ export const useUpdateMemory = () => {
         queryKey: ['memoryDetail', variables.id],
       });
     },
-    onError: (error) => {
-      message.error(t('message.error', { error: error.message }));
-    },
   });
 
   const updateMemory = useCallback(
diff --git a/web/src/pages/memory/memory-message/hook.ts b/web/src/pages/memory/memory-message/hook.ts
index e46bc2f87..899439c67 100644
--- a/web/src/pages/memory/memory-message/hook.ts
+++ b/web/src/pages/memory/memory-message/hook.ts
@@ -97,11 +97,13 @@ export const useMessageAction = () => {
           message_id: selectedMessageInfo.message_id,
           status: enable || false,
         })
-        .then(() => {
-          message.success(t('message.updated'));
-          queryClient.invalidateQueries({
-            queryKey: [MemoryApiAction.FetchMemoryMessage],
-          });
+        .then((data: any) => {
+          if (data.data.code === 0) {
+            message.success(t('message.updated'));
+            queryClient.invalidateQueries({
+              queryKey: [MemoryApiAction.FetchMemoryMessage],
+            });
+          }
         });
       setShowDeleteDialog(false);
     },