diff --git a/packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js b/packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js
index 38bb44727..7ed5555db 100644
--- a/packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js
+++ b/packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js
@@ -1,23 +1,6 @@
-import { expectTypeOf } from 'vitest';
-
 import { parser } from './parser/stateDiagram';
 import stateDb from './stateDb';
 import stateDiagram from './parser/stateDiagram.jison';
-import stateRendererV2 from './stateRenderer-v2';
-
-// Can use this instead of having to register diagrams and load/orchestrate them, etc.
-class FauxDiagramObj {
-  db = stateDb;
-  parser = parser;
-  renderer = stateRendererV2;
-
-  constructor(options = { db: stateDb, parser: parser, renderer: stateRendererV2 }) {
-    this.db = options.db;
-    this.parser = options.parser;
-    this.renderer = options.renderer;
-    this.parser.yy = this.db;
-  }
-}
 
 describe('state diagram V2, ', function () {
   // TODO - these examples should be put into ./parser/stateDiagram.spec.js
@@ -441,16 +424,4 @@ describe('state diagram V2, ', function () {
       expect(rel_Active_Active.relationTitle).toEqual('LOG');
     });
   });
-
-  describe('stateRenderer-v2', () => {
-    describe('getClasses', () => {
-      const diagramText = 'statediagram-v2\n';
-      const fauxStateDiagram = new FauxDiagramObj();
-
-      it('returns a {}', () => {
-        const result = stateRendererV2.getClasses(diagramText, fauxStateDiagram);
-        expectTypeOf(result).toBeObject();
-      });
-    });
-  });
 });
diff --git a/packages/mermaid/src/diagrams/state/stateRenderer-v2.spec.js b/packages/mermaid/src/diagrams/state/stateRenderer-v2.spec.js
new file mode 100644
index 000000000..3a118e607
--- /dev/null
+++ b/packages/mermaid/src/diagrams/state/stateRenderer-v2.spec.js
@@ -0,0 +1,31 @@
+import { expectTypeOf } from 'vitest';
+
+import { parser } from './parser/stateDiagram';
+import stateDb from './stateDb';
+import stateRendererV2 from './stateRenderer-v2';
+
+// Can use this instead of having to register diagrams and load/orchestrate them, etc.
+class FauxDiagramObj {
+  db = stateDb;
+  parser = parser;
+  renderer = stateRendererV2;
+
+  constructor(options = { db: stateDb, parser: parser, renderer: stateRendererV2 }) {
+    this.db = options.db;
+    this.parser = options.parser;
+    this.renderer = options.renderer;
+    this.parser.yy = this.db;
+  }
+}
+
+describe('stateRenderer-v2', () => {
+  describe('getClasses', () => {
+    const diagramText = 'statediagram-v2\n';
+    const fauxStateDiagram = new FauxDiagramObj();
+
+    it('returns a {}', () => {
+      const result = stateRendererV2.getClasses(diagramText, fauxStateDiagram);
+      expectTypeOf(result).toBeObject();
+    });
+  });
+});