diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index 2e1d37ea1..d2d561ae3 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -54,7 +54,7 @@ export default function ParameterComponent({
 			className="w-full flex flex-wrap justify-between items-center bg-gray-50 dark:bg-gray-800 dark:text-white mt-1 px-5 py-2"
 		>
 			<>
-				<div className="text-sm truncate">
+				<div className={"text-sm truncate w-full " + (left ? "" : "text-end")}>
 					{title}
 					<span className="text-red-600">{required ? " *" : ""}</span>
 				</div>
diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index 6e3aa6e67..0f9feeb7e 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -71,7 +71,7 @@ export default function GenericNode({
 							<div key={idx}>
 								{idx === 0 ? (
 									<div className="px-5 py-2 mt-2 dark:text-white text-center">
-										Inputs:
+										Inputs
 									</div>
 								) : (
 									<></>
@@ -101,7 +101,7 @@ export default function GenericNode({
 							</div>
 						))}
 					<div className="px-5 py-2 mt-2 dark:text-white text-center">
-						Output:
+						Output
 					</div>
 					<ParameterComponent
 						data={data}
@@ -109,7 +109,7 @@ export default function GenericNode({
 						title={data.type}
 						tooltipTitle={`Type: ${data.node.base_classes.join(" | ")}`}
 						id={[data.type, data.id, ...data.node.base_classes].join("|")}
-						type={"str"}
+						type={data.node.base_classes.join("|")}
 						left={false}
 					/>
 				</>