fixed selection which click after switching to a document

Signed-off-by: Ananya Sen <Ananya.Sen@motorola.com>

1 files changed, 7 insertions, 4 deletions

diff --git a/js/document/html-document.js b/js/document/html-document.js
index aac03606..9e3f0cdd 100755
--- a/js/document/html-document.js
+++ b/js/document/html-document.js
@@ -717,9 +717,7 @@ exports.HTMLDocument = Montage.create(TextDocument, {
             this.application.ninja.stage.drawUtils.gridHorizontalSpacing = this.gridHorizontalSpacing;
             this.application.ninja.stage.drawUtils.gridVerticalSpacing = this.gridVerticalSpacing;
 
-            if((typeof this.selectionModel !== 'undefined') && (this.selectionModel !== null)){
-                this.application.ninja.selectedElements = this.selectionModel.slice(0);
-            }
+
 
             if((this.savedLeftScroll!== null) && (this.savedTopScroll !== null)){
                 this.application.ninja.stage._iframeContainer.scrollLeft = this.savedLeftScroll;
@@ -727,12 +725,17 @@ exports.HTMLDocument = Montage.create(TextDocument, {
                 this.application.ninja.stage._iframeContainer.scrollTop = this.savedTopScroll;
                 this.application.ninja.stage._scrollLeft = this.savedTopScroll;
             }
-            this.application.ninja.stage.handleScroll();
+
+            if((typeof this.selectionModel !== 'undefined') && (this.selectionModel !== null)){
+                this.application.ninja.selectedElements = this.selectionModel.slice(0);
+            }
 
             this.application.ninja.appModel.show3dGrid = this.draw3DGrid;
 
             this.application.ninja.undocontroller.undoQueue = this.undoStack.slice(0);
             this.application.ninja.undocontroller.redoQueue = this.redoStack.slice(0);
+
+            this.application.ninja.currentSelectedContainer = this.documentRoot;
         }
     }
         ////////////////////////////////////////////////////////////////////