diff --git a/src/utils.spec.js b/src/utils.spec.js
index 4cab09b63..3446850da 100644
--- a/src/utils.spec.js
+++ b/src/utils.spec.js
@@ -137,9 +137,9 @@ describe('when cloning CSS ',function() {
         svg.setAttribute('id', 'mermaid-01');
 
         utils.cloneCssStyles(svg, { "default": { "styles": ["stroke:#fff","stroke-width:1.5px"] } });
-        expect(stylesToArray(svg)).toEqual([ '#mermaid-01 .node { stroke:#fff; stroke-width:1.5px; }' ]);
+        expect(stylesToArray(svg)).toEqual([ '#mermaid-01 .node>rect { stroke:#fff; stroke-width:1.5px; }' ]);
         // Also verify the elements around the styling
-        expect(svg.innerHTML).toBe('<style type="text/css" title="mermaid-svg-internal-css">/* <![CDATA[ */\n#mermaid-01 .node { stroke:#fff; stroke-width:1.5px; }\n/* ]]> */\n</style>');
+        expect(svg.innerHTML).toBe('<style type="text/css" title="mermaid-svg-internal-css">/* <![CDATA[ */\n#mermaid-01 .node>rect { stroke:#fff; stroke-width:1.5px; }\n/* ]]> */\n</style>');
     });
 
     it('should handle stylesheet in document with no classes in SVG', function () {
@@ -180,7 +180,7 @@ describe('when cloning CSS ',function() {
         var svg = generateSVG();
         addStyleToDocument();
         utils.cloneCssStyles(svg, { "default": { "styles": ["stroke:#fff","stroke-width:1.5px"] } });
-        expect(stylesToArray(svg)).toEqual([ '#mermaid-01 .node { stroke:#fff; stroke-width:1.5px; }', '.node { stroke: #eee; }', '.node-square { stroke: #bbb; }']);
+        expect(stylesToArray(svg)).toEqual([ '#mermaid-01 .node>rect { stroke:#fff; stroke-width:1.5px; }', '.node { stroke: #eee; }', '.node-square { stroke: #bbb; }']);
     });
 
     it('should handle a default class together with stylesheet in document and classDefs', function () {
@@ -189,11 +189,11 @@ describe('when cloning CSS ',function() {
         utils.cloneCssStyles(svg, { "default": { "styles": ["stroke:#fff","stroke-width:1.5px"] }, 
                                     "node-square": { "styles": ["fill:#eee", "stroke:#aaa"] },
                                     "node-circle": { "styles": ["fill:#444", "stroke:#111"] } });
-        expect(stylesToArray(svg)).toEqual([ '#mermaid-01 .node { stroke:#fff; stroke-width:1.5px; }', 
+        expect(stylesToArray(svg)).toEqual([ '#mermaid-01 .node>rect { stroke:#fff; stroke-width:1.5px; }',
                                              '.node { stroke: #eee; }', 
                                              '.node-square { stroke: #bbb; }',
-                                             '#mermaid-01 .node-square { fill:#eee; stroke:#aaa; }',
-                                             '#mermaid-01 .node-circle { fill:#444; stroke:#111; }'
+                                             '#mermaid-01 .node-square>rect { fill:#eee; stroke:#aaa; }',
+                                             '#mermaid-01 .node-circle>rect { fill:#444; stroke:#111; }'
                                              ]);
     });
 });