diff --git a/cypress/platform/current.html b/cypress/platform/current.html
index c341c70cf..cc4b9e862 100644
--- a/cypress/platform/current.html
+++ b/cypress/platform/current.html
@@ -38,12 +38,10 @@
           end
         a-->id1
       </div>
-      <div class="mermaid2 mermaid-apa" style="width: 100%; height: 20%;">
-        stateDiagram-v2
+      <div class="mermaid mermaid-apa" style="width: 100%; height: 20%;">
+        stateDiagram
         [*] --> Still
-        [*] --> Moving
         Still --> [*]
-        Moving --> [*]
       </div>
       <div class="mermaid2" style="width: 100%; height: 100%;">
         stateDiagram-v2
diff --git a/src/diagrams/state/stateDb.js b/src/diagrams/state/stateDb.js
index c4842cc60..d72e8f00c 100644
--- a/src/diagrams/state/stateDb.js
+++ b/src/diagrams/state/stateDb.js
@@ -3,7 +3,8 @@ import { logger } from '../../logger';
 let rootDoc = [];
 const setRootDoc = o => {
   logger.info('Setting root doc', o);
-  rootDoc = { id: 'root', doc: o };
+  // rootDoc = { id: 'root', doc: o };
+  rootDoc = o;
 };
 
 const getRootDoc = () => rootDoc;
@@ -26,17 +27,17 @@ const docTranslator = (parent, node, first) => {
   }
 };
 const getRootDocV2 = () => {
-  docTranslator({ id: 'root' }, rootDoc, true);
-  return rootDoc;
+  docTranslator({ id: 'root' }, { id: 'root', doc: rootDoc }, true);
+  return { id: 'root', doc: rootDoc };
 };
 
-const extract = root => {
+const extract = doc => {
   // const res = { states: [], relations: [] };
-  let doc = root.doc;
-  if (!doc) {
-    doc = root;
-  }
-  logger.info(root.doc);
+  // let doc = root.doc;
+  // if (!doc) {
+  //   doc = root;
+  // }
+  logger.info(doc);
   clear();
 
   doc.forEach(item => {