diff --git a/src/diagrams/class/styles.js b/src/diagrams/class/styles.js
index 43399af28..388fb3beb 100644
--- a/src/diagrams/class/styles.js
+++ b/src/diagrams/class/styles.js
@@ -12,7 +12,7 @@ const getStyles = options =>
 }
 
 .divider {
-  stroke: $nodeBorder;
+  stroke: ${options.nodeBorder};
   stroke: 1;
 }
 
diff --git a/src/styles.js b/src/styles.js
index 810760bdb..06769dde6 100644
--- a/src/styles.js
+++ b/src/styles.js
@@ -14,7 +14,8 @@ const themes = {
   'flowchart-v2': flowchart,
   sequence,
   gantt,
-  class: classDiagram,
+  classDiagram,
+  'classDiagram-v2': classDiagram,
   stateDiagram,
   state: stateDiagram,
   git,