diff --git a/packages/mermaid/src/diagrams/class/classDiagram-v2.ts b/packages/mermaid/src/diagrams/class/classDiagram-v2.ts
index fb6c77a47..ec5398d29 100644
--- a/packages/mermaid/src/diagrams/class/classDiagram-v2.ts
+++ b/packages/mermaid/src/diagrams/class/classDiagram-v2.ts
@@ -1,13 +1,13 @@
 import type { DiagramDefinition } from '../../diagram-api/types.js';
 // @ts-ignore: JISON doesn't support types
 import parser from './parser/classDiagram.jison';
-import classDb from './classDb.js';
+import db from './classDb.js';
 import styles from './styles.js';
 import renderer from './classRenderer-v2.js';
 
 export const diagram: DiagramDefinition = {
   parser,
-  db: classDb,
+  db,
   renderer,
   styles,
   init: (cnf) => {
@@ -15,6 +15,6 @@ export const diagram: DiagramDefinition = {
       cnf.class = {};
     }
     cnf.class.arrowMarkerAbsolute = cnf.arrowMarkerAbsolute;
-    classDb.clear();
+    db.clear();
   },
 };
diff --git a/packages/mermaid/src/diagrams/class/classDiagram.ts b/packages/mermaid/src/diagrams/class/classDiagram.ts
index 25a4abc63..7f027c186 100644
--- a/packages/mermaid/src/diagrams/class/classDiagram.ts
+++ b/packages/mermaid/src/diagrams/class/classDiagram.ts
@@ -1,13 +1,13 @@
 import type { DiagramDefinition } from '../../diagram-api/types.js';
 // @ts-ignore: JISON doesn't support types
 import parser from './parser/classDiagram.jison';
-import classDb from './classDb.js';
+import db from './classDb.js';
 import styles from './styles.js';
 import renderer from './classRenderer.js';
 
 export const diagram: DiagramDefinition = {
   parser,
-  db: classDb,
+  db,
   renderer,
   styles,
   init: (cnf) => {
@@ -15,6 +15,6 @@ export const diagram: DiagramDefinition = {
       cnf.class = {};
     }
     cnf.class.arrowMarkerAbsolute = cnf.arrowMarkerAbsolute;
-    classDb.clear();
+    db.clear();
   },
 };