diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index ee62461bd..3b5ffc15d 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -242,22 +242,21 @@ export default function GenericNode({
             >
               {" "}
             </div>
-            {/* <div className="px-5 py-2 mt-2 text-center">
-                  Output
-              </div> */}
-            <ParameterComponent
-              data={data}
-              color={nodeColors[types[data.type]] ?? nodeColors.unknown}
-              title={
-                data.node.output_types && data.node.output_types.length > 0
-                  ? data.node.output_types.join("|")
-                  : data.type
-              }
-              tooltipTitle={data.node.base_classes.join("\n")}
-              id={[data.type, data.id, ...data.node.base_classes].join("|")}
-              type={data.node.base_classes.join("|")}
-              left={false}
-            />
+            {data.node.base_classes?.length > 0 && (
+              <ParameterComponent
+                data={data}
+                color={nodeColors[types[data.type]] ?? nodeColors.unknown}
+                title={
+                  data.node.output_types && data.node.output_types.length > 0
+                    ? data.node.output_types.join("|")
+                    : data.type
+                }
+                tooltipTitle={data.node.base_classes.join("\n")}
+                id={[data.type, data.id, ...data.node.base_classes].join("|")}
+                type={data.node.base_classes.join("|")}
+                left={false}
+              />
+            )}
           </>
         </div>
       </div>