diff --git a/packages/mermaid/src/diagrams/class/classDiagram.spec.ts b/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
index 19e31e6d2..532c8aaa7 100644
--- a/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
+++ b/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
@@ -814,19 +814,17 @@ describe('given a class diagram with members and methods ', function () {
     });
 
     it('should handle direct member declaration', function () {
-      const str = 'classDiagram\n' + 'Car : wheels';
-
-      parser.parse(str);
-      expect(classDb.getClasses()).toHaveProperty('Car');
-      expect(classDb.getClasses()['Car']['members']).toContain('wheels');
+      parser.parse('classDiagram\n' + 'Car : wheels');
+      const car = classDb.getClass('Car');
+      expect(car.members.length).toBe(1);
+      expect(car.members[0].id).toBe('wheels');
     });
 
     it('should handle direct member declaration with type', function () {
-      const str = 'classDiagram\n' + 'Car : int wheels';
-
-      parser.parse(str);
-      expect(classDb.getClasses()).toHaveProperty('Car');
-      expect(classDb.getClasses()['Car']['members']).toContain('int wheels');
+      parser.parse('classDiagram\n' + 'Car : int wheels');
+      const car = classDb.getClass('Car');
+      expect(car.members.length).toBe(1);
+      expect(car.members[0].id).toBe('int wheels');
     });
 
     it('should handle simple member declaration with type', function () {