diff --git a/src/dagre-wrapper/intersect/intersect-rect.js b/src/dagre-wrapper/intersect/intersect-rect.js
index 9847f9897..f8c03a2af 100644
--- a/src/dagre-wrapper/intersect/intersect-rect.js
+++ b/src/dagre-wrapper/intersect/intersect-rect.js
@@ -2,7 +2,6 @@ const intersectRect = (node, point) => {
   var x = node.x;
   var y = node.y;
 
-  console.log(node, point);
   // Rectangle intersection algorithm from:
   // http://math.stackexchange.com/questions/108113/find-edge-between-two-boxes
   var dx = point.x - x;
diff --git a/src/diagrams/state/stateDb.js b/src/diagrams/state/stateDb.js
index 9b8031f75..c4842cc60 100644
--- a/src/diagrams/state/stateDb.js
+++ b/src/diagrams/state/stateDb.js
@@ -30,8 +30,13 @@ const getRootDocV2 = () => {
   return rootDoc;
 };
 
-const extract = doc => {
+const extract = root => {
   // const res = { states: [], relations: [] };
+  let doc = root.doc;
+  if (!doc) {
+    doc = root;
+  }
+  logger.info(root.doc);
   clear();
 
   doc.forEach(item => {