1 files changed, 3 insertions, 3 deletions
diff --git a/js/mediators/element-mediator.js b/js/mediators/element-mediator.js
index 7657112f..88baf3d9 100755
--- a/js/mediators/element-mediator.js
+++ b/js/mediators/element-mediator.js
@@ -54,7 +54,7 @@ exports.ElementMediator = Montage.create(Component, {
            document.application.undoManager.add(undoLabel, this.removeElements, this, elements, notify);
-            this.application.ninja.documentController.activeDocument.model.needsSave = true;
+            this.application.ninja.currentDocument.model.needsSave = true;
            if(notify || notify === undefined) {
                NJevent("elementAdded", elements);
@@ -83,7 +83,7 @@ exports.ElementMediator = Montage.create(Component, {
            document.application.undoManager.add(undoLabel, this.addElements, this, elements, null, notify);
-            this.application.ninja.documentController.activeDocument.model.needsSave = true;
+            this.application.ninja.currentDocument.model.needsSave = true;
            NJevent("elementsRemoved", elements);
        }
@@ -98,7 +98,7 @@ exports.ElementMediator = Montage.create(Component, {
            document.application.undoManager.add(undoLabel, this.replaceElement, this, oldChild, newChild);
-            this.application.ninja.documentController.activeDocument.model.needsSave = true;
+            this.application.ninja.currentDocument.model.needsSave = true;
            if(notify || notify === undefined) {
                NJevent("elementReplaced", {type : "replaceElement", data: {"newChild": newChild, "oldChild": oldChild}});

diff --git a/js/mediators/element-mediator.js b/js/mediators/element-mediator.js index 7657112f..88baf3d9 100755 --- a/js/mediators/element-mediator.js +++ b/js/mediators/element-mediator.js
@@ -54,7 +54,7 @@ exports.ElementMediator = Montage.create(Component, {
54		54
55	document.application.undoManager.add(undoLabel, this.removeElements, this, elements, notify);	55	document.application.undoManager.add(undoLabel, this.removeElements, this, elements, notify);
56		56
57	this.application.ninja.documentController.activeDocument.model.needsSave = true;	57	this.application.ninja.currentDocument.model.needsSave = true;
58		58
59	if(notify \|\| notify === undefined) {	59	if(notify \|\| notify === undefined) {
60	NJevent("elementAdded", elements);	60	NJevent("elementAdded", elements);
@@ -83,7 +83,7 @@ exports.ElementMediator = Montage.create(Component, {
83		83
84	document.application.undoManager.add(undoLabel, this.addElements, this, elements, null, notify);	84	document.application.undoManager.add(undoLabel, this.addElements, this, elements, null, notify);
85		85
86	this.application.ninja.documentController.activeDocument.model.needsSave = true;	86	this.application.ninja.currentDocument.model.needsSave = true;
87		87
88	NJevent("elementsRemoved", elements);	88	NJevent("elementsRemoved", elements);
89	}	89	}
@@ -98,7 +98,7 @@ exports.ElementMediator = Montage.create(Component, {
98		98
99	document.application.undoManager.add(undoLabel, this.replaceElement, this, oldChild, newChild);	99	document.application.undoManager.add(undoLabel, this.replaceElement, this, oldChild, newChild);
100		100
101	this.application.ninja.documentController.activeDocument.model.needsSave = true;	101	this.application.ninja.currentDocument.model.needsSave = true;
102		102
103	if(notify \|\| notify === undefined) {	103	if(notify \|\| notify === undefined) {
104	NJevent("elementReplaced", {type : "replaceElement", data: {"newChild": newChild, "oldChild": oldChild}});	104	NJevent("elementReplaced", {type : "replaceElement", data: {"newChild": newChild, "oldChild": oldChild}});