diff --git a/src/frontend/src/CustomNodes/GenericNode/components/NodeInputField/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/NodeInputField/index.tsx
index ad5cc66f5..a1d1f977b 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/NodeInputField/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/NodeInputField/index.tsx
@@ -108,8 +108,8 @@ export default function NodeInputField({
-
0 || showHiddenOutputs)
- }
+ lastInput={!(shownOutputs.length > 0 || showHiddenOutputs)}
key={memoizedKey}
data={data}
colors={memoizedColor.colors}
diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index 40bb9ebd7..dc3af464a 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -48,11 +48,11 @@ const HiddenOutputsButton = memo(
}) => (