diff --git a/src/plugins/wizard/common/wizard_saved_object_attributes.ts b/src/plugins/wizard/common/wizard_saved_object_attributes.ts
index 1dc740d68637..e0e1e1cb6de7 100644
--- a/src/plugins/wizard/common/wizard_saved_object_attributes.ts
+++ b/src/plugins/wizard/common/wizard_saved_object_attributes.ts
@@ -13,4 +13,7 @@ export interface WizardSavedObjectAttributes extends SavedObjectAttributes {
   visualizationState?: string;
   styleState?: string;
   version: number;
+  searchSourceFields?: {
+    index?: string;
+  };
 }
diff --git a/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx b/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx
index bc2d4548dda1..1acbd295e43a 100644
--- a/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx
+++ b/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx
@@ -116,8 +116,15 @@ export class WizardEmbeddable extends Embeddable<SavedObjectEmbeddableInput, Wiz
       return;
     }
     const { visualization, style } = this.serializedState;
+
+    const vizStateWithoutIndex = JSON.parse(visualization);
+    const visualizationState = {
+      searchField: vizStateWithoutIndex.searchField,
+      activeVisualization: vizStateWithoutIndex.activeVisualization,
+      indexPattern: this.savedWizard?.searchSourceFields?.index,
+    };
     const rootState = {
-      visualization: JSON.parse(visualization),
+      visualization: visualizationState,
       style: JSON.parse(style),
     };
     const visualizationName = rootState.visualization?.activeVisualization?.name ?? '';