diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index e9f6711ae..988b9b964 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -164,7 +164,7 @@ export default function GenericNode({
                   <span className="text-lg">{data?.node?.icon}</span>
                 ) : (
                   <IconComponent
-                    name={data.node?.flow ? "group_components" : name}
+                    name={data?.node?.icon ? data?.node?.icon : name}
                     className={
                       "generic-node-icon " +
                       (!showNode ? "absolute inset-x-6 h-12 w-12" : "")