diff --git a/web/app/components/workflow/hooks/use-workflow.ts b/web/app/components/workflow/hooks/use-workflow.ts
index 0f6ae59b6e..f56c5a615d 100644
--- a/web/app/components/workflow/hooks/use-workflow.ts
+++ b/web/app/components/workflow/hooks/use-workflow.ts
@@ -608,7 +608,11 @@ export const useToolIcon = (data: Node['data']) => {
         targetTools = customTools
       else
         targetTools = workflowTools
-      return targetTools.find(toolWithProvider => toolWithProvider.id === data.provider_id)?.icon
+      return targetTools.find((toolWithProvider) => {
+        return toolWithProvider.id === data.provider_id
+        || toolWithProvider.id === `langgenius/${data.provider_id}/${data.provider_id}`
+        || toolWithProvider.id === `langgenius/${data.provider_id}_tool/${data.provider_id}`
+      })?.icon
     }
   }, [data, buildInTools, customTools, workflowTools])