1 files changed, 4 insertions, 4 deletions

diff --git a/js/controllers/document-controller.js b/js/controllers/document-controller.js
index 588481bb..ee7ca82c 100755
--- a/js/controllers/document-controller.js
+++ b/js/controllers/document-controller.js
@@ -68,7 +68,7 @@ exports.DocumentController = Montage.create(Component, {
                 return;
             }
 
-            if(this._currentDocument) {
+            if(this._currentDocument && this._currentDocument.model.currentView) {
                 this._currentDocument.model.currentView.hide();
             }
 
@@ -80,12 +80,12 @@ exports.DocumentController = Montage.create(Component, {
             } else if(this._currentDocument.currentView === "design") {
                 document.getElementById("codeViewContainer").style.display = "none";
                 document.getElementById("iframeContainer").style.display = "block";
-                this._currentDocument.model.currentView.show();
+                if (this._currentDocument.model.currentView) this._currentDocument.model.currentView.show();
                 this._currentDocument.model.views.design._liveNodeList = this._currentDocument.model.documentRoot.getElementsByTagName('*');
             } else {
                 document.getElementById("iframeContainer").style.display = "none";
                 this._currentDocument.model.parentContainer.style["display"] = "block";
-                this._currentDocument.model.currentView.show();
+                if (this._currentDocument.model.currentView) this._currentDocument.model.currentView.show();
             }
 
         }
@@ -356,7 +356,7 @@ exports.DocumentController = Montage.create(Component, {
                     break;
                 default:
                     //Open in code view
-                    Montage.create(TextDocument).init(file, this.application.ninja, this.application.ninja.openDocument, 'code');
+                    Montage.create(TextDocument).init(file, this.application.ninja, this.application.ninja.openDocument, 'code', document.getElementById("codeViewContainer"));
                     break;
             }
         }