diff --git a/packages/mermaid/src/rendering-util/rendering-elements/shapes/drawRect.ts b/packages/mermaid/src/rendering-util/rendering-elements/shapes/drawRect.ts
index 2efda3fa2..bef59decc 100644
--- a/packages/mermaid/src/rendering-util/rendering-elements/shapes/drawRect.ts
+++ b/packages/mermaid/src/rendering-util/rendering-elements/shapes/drawRect.ts
@@ -55,7 +55,7 @@ export const drawRect = async (parent: SVGAElement, node: Node, options: RectOpt
     }
 
     rect
-      .attr('class', 'basic label-container')
+      .attr('class', rectClass)
       .attr('style', nodeStyles)
       .attr('rx', rx)
       .attr('data-id', 'abc')