From d9827337b0d7286dfdd82dec88e713613a4f36e5 Mon Sep 17 00:00:00 2001
From: Rares Mardare <rares.mardare@grafana.com>
Date: Wed, 5 Oct 2022 12:32:52 +0300
Subject: [PATCH] pick override property if isOverridePreview

---
 grafana-plugin/src/models/schedule/schedule.helpers.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/grafana-plugin/src/models/schedule/schedule.helpers.ts b/grafana-plugin/src/models/schedule/schedule.helpers.ts
index b97aa545..c025e249 100644
--- a/grafana-plugin/src/models/schedule/schedule.helpers.ts
+++ b/grafana-plugin/src/models/schedule/schedule.helpers.ts
@@ -67,7 +67,7 @@ export const getShiftsFromStore = (
   const source = isOverridePreview ? store.scheduleStore.overridePreview : store.scheduleStore.finalPreview;
   return source
     ? store.scheduleStore.finalPreview
-    : (store.scheduleStore.events[scheduleId]?.['final']?.[getFromString(startMoment)] as any);
+    : (store.scheduleStore.events[scheduleId]?.[isOverridePreview ? 'override' : 'final']?.[getFromString(startMoment)] as any);
 };
 
 export const getLayersFromStore = (store: RootStore, scheduleId: Schedule['id'], startMoment: dayjs.Dayjs): Layer[] => {
@@ -92,7 +92,7 @@ export const getOverridesFromStore = (
   const source = isOverridePreview ? store.scheduleStore.overridePreview : store.scheduleStore.rotationPreview;
   return source
     ? store.scheduleStore.rotationPreview
-    : (store.scheduleStore.events[scheduleId]?.['rotation']?.[getFromString(startMoment)] as Layer[]);
+    : (store.scheduleStore.events[scheduleId]?.[isOverridePreview ? 'override' : 'rotation']?.[getFromString(startMoment)] as Layer[]);
 };
 
 export const splitToLayers = (