diff --git a/src/diagrams/class/classRenderer.js b/src/diagrams/class/classRenderer.js
index 5e6208ac5..4e7128819 100644
--- a/src/diagrams/class/classRenderer.js
+++ b/src/diagrams/class/classRenderer.js
@@ -570,8 +570,8 @@ export const draw = function(text, id) {
     }
   });
 
-  diagram.attr('height', '100%');
-  diagram.attr('width', `${g.graph().width * 1.5 + 20}`);
+  diagram.attr('height', g.graph().height + 40);
+  diagram.attr('width', g.graph().width * 1.5 + 20);
   diagram.attr('viewBox', '-10 -10 ' + (g.graph().width + 20) + ' ' + (g.graph().height + 20));
 };