Fix unassigned track element on document switch

Signed-off-by: Jonathan Duran <jduran@motorola.com>

1 files changed, 4 insertions, 0 deletions

diff --git a/js/panels/Timeline/TimelineTrack.reel/TimelineTrack.js b/js/panels/Timeline/TimelineTrack.reel/TimelineTrack.js
index 18df20c2..c6b67c9c 100644
--- a/js/panels/Timeline/TimelineTrack.reel/TimelineTrack.js
+++ b/js/panels/Timeline/TimelineTrack.reel/TimelineTrack.js
@@ -394,6 +394,10 @@ var TimelineTrack = exports.TimelineTrack = Montage.create(Component, {
     draw:{
         value:function () {
             this.ninjaStylesContoller = this.application.ninja.stylesController;
+            var selectedIndex = this.application.ninja.timeline.getLayerIndexByID(this.trackID);
+            if(this.application.ninja.timeline.arrLayers[selectedIndex].layerData.elementsList[0]){
+                this.animatedElement = this.application.ninja.timeline.arrLayers[selectedIndex].layerData.elementsList[0];
+            }
         }
     },