editor: minor refactoring

[fnpeditor.git] / src / editor / modules / documentCanvas / canvas / documentElement.js

diff --git a/src/editor/modules/documentCanvas/canvas/documentElement.js b/src/editor/modules/documentCanvas/canvas/documentElement.js
index 8a5193b..50a188a 100644 (file)
--- a/src/editor/modules/documentCanvas/canvas/documentElement.js
+++ b/src/editor/modules/documentCanvas/canvas/documentElement.js
@@ -18,6 +18,7 @@ var DocumentElement = function(wlxmlNode, canvas) {
 
     this.dom = this.createDOM();
     this.dom.data('canvas-element', this);
+    this.wlxmlNode.setData('canvasElement', this);
 };
 
 $.extend(DocumentElement.prototype, {
@@ -89,7 +90,6 @@ $.extend(DocumentElement.prototype, {
 // DocumentNodeElement represents an element node from WLXML document rendered inside Canvas
 var DocumentNodeElement = function(wlxmlNode, canvas) {
     DocumentElement.call(this, wlxmlNode, canvas);
-    wlxmlNode.setData('canvasElement', this);
     this.init(this.dom);
 };
 
@@ -154,6 +154,9 @@ $.extend(DocumentNodeElement.prototype, {
     detach: function(isChild) {
         var parents;
 
+        if(this.gutterGroup) {
+            this.gutterGroup.remove();
+        }
         if(_.isFunction(this.children)) {
             this.children().forEach(function(child) {
                 child.detach(true);