diff --git a/src/routes/Edit.svelte b/src/routes/Edit.svelte
index 98c11cb3..303fbe6e 100644
--- a/src/routes/Edit.svelte
+++ b/src/routes/Edit.svelte
@@ -88,7 +88,7 @@
             `;
                 break;
             case "StateDiagram":
-                code = `stateDiagram
+                code = `stateDiagram-v2
     [*] --> Still
     Still --> [*]
     Still --> Moving