From 63e522bf822d48f397e91b70d54444761eaa5cfa Mon Sep 17 00:00:00 2001
From: icarus <eurfelux@gmail.com>
Date: Wed, 22 Oct 2025 06:22:58 +0800
Subject: [PATCH] fix(CodeToolbar): correct tool cleanup and fix callback
 invocation

Ensure the correct tool ID is used for cleanup and wrap fixCode in arrow function to maintain context
---
 .../src/components/CodeToolbar/hooks/useMermaidFixTool.tsx    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/components/CodeToolbar/hooks/useMermaidFixTool.tsx b/src/renderer/src/components/CodeToolbar/hooks/useMermaidFixTool.tsx
index b30e8a03c6..1a7c17b655 100644
--- a/src/renderer/src/components/CodeToolbar/hooks/useMermaidFixTool.tsx
+++ b/src/renderer/src/components/CodeToolbar/hooks/useMermaidFixTool.tsx
@@ -169,10 +169,10 @@ export const useMermaidFixTool = ({ enabled, context, onSave, setError, setTools
         icon: <WrenchIcon size={'1rem'} className="tool-icon" />,
         tooltip: t('code_block.mermaid_fix.label'),
         visible: () => error !== undefined && error !== null,
-        onClick: fixCode
+        onClick: () => fixCode()
       })
     }
 
-    return () => removeTool(TOOL_SPECS.expand.id)
+    return () => removeTool(TOOL_SPECS.mermaid_fix.id)
   }, [enabled, error, fixCode, registerTool, removeTool, t])
 }