From 70aed8b47f2fc3436109480f64b1da1e51860417 Mon Sep 17 00:00:00 2001
From: Nathan Reese <reese.nathan@gmail.com>
Date: Wed, 2 Apr 2025 12:27:38 -0600
Subject: [PATCH] [dashboard][8.17][8.16] fix State being dropped when editing
 visualize embeddables

---
 .../dashboard_container/embeddable/dashboard_container.tsx   | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/plugins/dashboard/public/dashboard_container/embeddable/dashboard_container.tsx b/src/plugins/dashboard/public/dashboard_container/embeddable/dashboard_container.tsx
index b3ae21d4635f3..8912f8ba69c2c 100644
--- a/src/plugins/dashboard/public/dashboard_container/embeddable/dashboard_container.tsx
+++ b/src/plugins/dashboard/public/dashboard_container/embeddable/dashboard_container.tsx
@@ -342,7 +342,10 @@ export class DashboardContainer
     this.hasOverlays$ = dashboardApi.hasOverlays$;
     this.openOverlay = dashboardApi.openOverlay;
     this.hasRunMigrations$ = dashboardApi.hasRunMigrations$;
-    this.setLastSavedInput = dashboardApi.setLastSavedInput;
+    this.setLastSavedInput = (lastSavedInput: DashboardContainerInput) => {
+      dashboardApi.setLastSavedInput(lastSavedInput);
+      this.setPanels(lastSavedInput.panels);
+    };
     this.lastSavedInput$ = dashboardApi.lastSavedInput$;
     this.savedObjectId = dashboardApi.savedObjectId;
     this.setSavedObjectId = dashboardApi.setSavedObjectId;