diff --git a/src/frontend/src/CustomNodes/GenericNode/components/NodeStatus/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/NodeStatus/index.tsx
index 5951b7935..2ca06afcd 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/NodeStatus/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/NodeStatus/index.tsx
@@ -71,9 +71,6 @@ export default function NodeStatus({
     return frozen ? frozenClass : className;
   };
 
-  const getNodeSizeClass = (showNode) =>
-    showNode ? "w-96 rounded-lg" : "w-26 h-26 rounded-full";
-
   const getNodeBorderClassName = (
     selected: boolean,
     showNode: boolean,
@@ -87,10 +84,8 @@ export default function NodeStatus({
     );
 
     const baseBorderClass = getBaseBorderClass(selected);
-    const nodeSizeClass = getNodeSizeClass(showNode);
     const names = classNames(
       baseBorderClass,
-      nodeSizeClass,
       "generic-node-div group/node",
       specificClassFromBuildStatus,
     );
diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index bcbd2ce5e..25dc6c91f 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -287,7 +287,12 @@ export default function GenericNode({
   return (
     <>
       {memoizedNodeToolbarComponent}
-      <div className={borderColor}>
+      <div
+        className={cn(
+          borderColor,
+          showNode ? "w-96 rounded-lg" : "w-26 h-26 rounded-full",
+        )}
+      >
         {data.node?.beta && showNode && (
           <div className="beta-badge-wrapper">
             <div className="beta-badge-content">BETA</div>