Adding back the stage references in the 3d classes.

Signed-off-by: Nivesh Rajbhandari <mqg734@motorola.com>

1 files changed, 4 insertions, 0 deletions

diff --git a/js/stage/stage.reel/stage.js b/js/stage/stage.reel/stage.js
index 4505ff66..dc00eac2 100755
--- a/js/stage/stage.reel/stage.js
+++ b/js/stage/stage.reel/stage.js
@@ -1090,7 +1090,11 @@ exports.Stage = Montage.create(Component, {
 
             workingPlane = [0,0,1,0];
 
+            this.viewUtils.setStageElement(this.application.ninja.currentDocument.model.documentRoot);
+            this.viewUtils.setRootElement(this.application.ninja.currentDocument.model.documentRoot.parentNode);
+
             this.snapManager._isCacheInvalid = true;
+            this.snapManager.currentStage = this.application.ninja.currentDocument.model.documentRoot;
             this.snapManager.setupDragPlaneFromPlane (workingPlane);
 
             this.drawUtils.initializeFromDocument();