diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index fe427dc97..95fd70a7e 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -95,15 +95,11 @@ export default function GenericNode({
           "generic-node-div"
         )}
       >
-        <div className="absolute right-0 top-0">
-          {data.node.beta && (
-            <div className="h-16 w-16 overflow-hidden rounded-tr-lg">
-              <div className="mt-2 w-24 rotate-45 bg-status-blue text-center text-xs font-semibold text-destructive-foreground">
-                BETA
-              </div>
-            </div>
-          )}
-        </div>
+        {data.node.beta && (
+          <div className="beta-badge-wrapper">
+            <div className="beta-badge-content">BETA</div>
+          </div>
+        )}
         <div className="generic-node-div-title">
           <div className="generic-node-title-arrangement">
             <IconComponent
diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index bb32e1dc8..c882ab6ea 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -1111,7 +1111,12 @@ The cursor: default; property value restores the browser's default cursor style
   .ace-editor-save-btn {
     @apply flex-max-width h-fit justify-end;
   }
-
+  .beta-badge-wrapper {
+    @apply absolute right-0 top-0 h-16 w-16 overflow-hidden rounded-tr-lg;
+  }
+  .beta-badge-content {
+    @apply mt-2 w-24 rotate-45 bg-status-blue text-center text-xs font-semibold text-destructive-foreground
+  }
   .export-modal-save-api {
     @apply font-medium leading-none peer-disabled:cursor-not-allowed peer-disabled:opacity-70;
   }