From 9ed9a9dd49087f40a532c78cc34fbe3d69f574bc Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Wed, 29 Nov 2023 14:01:26 -0300
Subject: [PATCH] Fixed icon not colorful on nodes

---
 .../src/CustomNodes/GenericNode/index.tsx         |  2 +-
 .../src/components/genericIconComponent/index.tsx | 15 ++++++++++-----
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index afa2558f7..9a6f818f2 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -176,7 +176,7 @@ export default function GenericNode({
               <IconComponent
                 name={data.node?.flow ? "Ungroup" : name}
                 className={
-                  "generic-node-icon " +
+                  "generic-node-icon" +
                   (!showNode && "absolute inset-x-6 h-12 w-12")
                 }
                 iconColor={`${nodeColors[types[data.type]]}`}
diff --git a/src/frontend/src/components/genericIconComponent/index.tsx b/src/frontend/src/components/genericIconComponent/index.tsx
index db415ed5b..d040cbe7e 100644
--- a/src/frontend/src/components/genericIconComponent/index.tsx
+++ b/src/frontend/src/components/genericIconComponent/index.tsx
@@ -8,14 +8,19 @@ const ForwardedIconComponent = forwardRef(
     ref
   ) => {
     const TargetIcon = nodeIconsLucide[name] ?? nodeIconsLucide["unknown"];
+
+    const style = {
+      strokeWidth: 1.5,
+      className: className,
+      ...(stroke && { stroke: stroke }),
+      ...(iconColor && { color: iconColor, stroke: stroke }),
+    };
+
     return (
       <TargetIcon
-        strokeWidth={1.5}
-        className={className}
-        style={iconColor ? { color: iconColor } : {}}
+        style={style}
         ref={ref}
-        stroke={stroke ? stroke : "currentColor"}
-        data-testid={id ? `${id}-${name}` : "icon-" + name}
+        data-testid={id ? `${id}-${name}` : `icon-${name}`}
       />
     );
   }