Refactor: Return without json

1 files changed, 8 insertions, 3 deletions

diff --git a/internal/fsm/fsm.go b/internal/fsm/fsm.go
index 63c9ac2..292e09e 100644
--- a/internal/fsm/fsm.go
+++ b/internal/fsm/fsm.go
@@ -1,9 +1,9 @@
 package fsm
 
 import (
+	"fmt"
 	"os"
 	"os/exec"
-	"fmt"
 	"path"
 	"sort"
 )
@@ -149,7 +149,13 @@ func (fsm *FSM) generateMermaidGraph() string {
 			} else {
 				graph += "\nstyle " + fsm.GetName(state) + " fill:white\n"
 			}
-			graph += fsm.GetName(state) + "(" + fsm.GetName(state) + ") " + "-->|" + transition.Description + "| " + fsm.GetName(transition.To) + "\n"
+			graph += fsm.GetName(
+				state,
+			) + "(" + fsm.GetName(
+				state,
+			) + ") " + "-->|" + transition.Description + "| " + fsm.GetName(
+				transition.To,
+			) + "\n"
 		}
 	}
 	return graph
@@ -162,4 +168,3 @@ func (fsm *FSM) GenerateGraph() string {
 
 	return ""
 }
-