From 8435330534455a0e46a498abb4df0ac83ec0c980 Mon Sep 17 00:00:00 2001
From: Justin Greywolf <jgreywolf@users.noreply.github.com>
Date: Sun, 2 Jul 2023 18:10:53 -0700
Subject: [PATCH] Update packages/mermaid/src/diagrams/class/classTypes.ts

Co-authored-by: Alois Klink <alois@aloisklink.com>
---
 packages/mermaid/src/diagrams/class/classTypes.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/mermaid/src/diagrams/class/classTypes.ts b/packages/mermaid/src/diagrams/class/classTypes.ts
index 2835ce08a..5eaf4fba8 100644
--- a/packages/mermaid/src/diagrams/class/classTypes.ts
+++ b/packages/mermaid/src/diagrams/class/classTypes.ts
@@ -19,13 +19,13 @@ export interface ClassNode {
 export class ClassMember {
   id!: string;
   cssStyle!: string;
-  memberType!: string;
+  memberType!: 'method' | 'attribute';
   visibility!: string;
   classifier!: string;
   parameters!: string;
   returnType!: string;
 
-  constructor(input: string, memberType: string) {
+  constructor(input: string, memberType: 'method' | 'attribute') {
     this.memberType = memberType;
     this.visibility = '';
     this.classifier = '';