2 files changed, 6 insertions, 4 deletions

diff --git a/js/document/html-document.js b/js/document/html-document.js
index 16595d14..ae2b4600 100755
--- a/js/document/html-document.js
+++ b/js/document/html-document.js
@@ -235,7 +235,7 @@ exports.HTMLDocument = Montage.create(TextDocument, {
         //
     initialize: {
                 value: function(file, uuid, iframe, callback) {
-                        //console.log('allow');
+                        this.application.ninja.documentController._hackRootFlag = false;
                         //
                         this._userDocument = file;
                         //
@@ -363,7 +363,8 @@ exports.HTMLDocument = Montage.create(TextDocument, {
         //
     handleEvent: {
         value: function(event){
-                        
+                this.application.ninja.documentController._hackRootFlag = true;
+                        //console.log(this._userDocument.root, this);
                 //TODO: Clean up, using for prototyping save
                 this._templateDocument = {};
                 this._templateDocument.head = this.iframe.contentWindow.document.getElementById("userHead");;
diff --git a/js/document/text-document.js b/js/document/text-document.js
index 6f8efaad..3e20e1f6 100755
--- a/js/document/text-document.js
+++ b/js/document/text-document.js
@@ -6,9 +6,10 @@ No rights, expressed or implied, whatsoever to this software are provided by Mot
 
 //BaseDocument Object for all files types and base class for HTML documents.
 
-var Montage = require("montage/core/core").Montage;
+var Montage =   require("montage/core/core").Montage,
+        Component = require("montage/ui/component").Component;
 
-var TextDocument = exports.TextDocument = Montage.create(Montage, {
+var TextDocument = exports.TextDocument = Montage.create(Component, {
         
         
         //TODO: Clean up, test