diff --git a/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx b/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx
index bc2d4548dda1..e1836bd65402 100644
--- a/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx
+++ b/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx
@@ -29,6 +29,7 @@ import {
 } from '../../../data/public';
 import { validateSchemaState } from '../application/utils/validate_schema_state';
 import { getExpressionLoader, getTypeService } from '../plugin_services';
+import { PersistedState } from '../../../visualizations/public';
 
 // Apparently this needs to match the saved object type for the clone and replace panel actions to work
 export const WIZARD_EMBEDDABLE = WIZARD_SAVED_OBJECT;
@@ -65,6 +66,7 @@ export class WizardEmbeddable extends Embeddable<SavedObjectEmbeddableInput, Wiz
   private node?: HTMLElement;
   private savedWizard?: WizardSavedObjectAttributes;
   private serializedState?: { visualization: string; style: string };
+  private uiState?: PersistedState;
 
   constructor(
     timefilter: TimefilterContract,
@@ -90,6 +92,7 @@ export class WizardEmbeddable extends Embeddable<SavedObjectEmbeddableInput, Wiz
     );
 
     this.savedWizard = savedWizard;
+    this.uiState = new PersistedState();
 
     this.autoRefreshFetchSubscription = timefilter
       .getAutoRefreshFetch$()
@@ -219,6 +222,7 @@ export class WizardEmbeddable extends Embeddable<SavedObjectEmbeddableInput, Wiz
         query: this.input.query,
         filters: this.input.filters,
       },
+      uiState: this.uiState,
     };
     if (this.abortController) {
       this.abortController.abort();
diff --git a/src/plugins/wizard/public/embeddable/wizard_embeddable_factory.tsx b/src/plugins/wizard/public/embeddable/wizard_embeddable_factory.tsx
index f2d92d001303..8a18472f6b44 100644
--- a/src/plugins/wizard/public/embeddable/wizard_embeddable_factory.tsx
+++ b/src/plugins/wizard/public/embeddable/wizard_embeddable_factory.tsx
@@ -71,11 +71,8 @@ export class WizardEmbeddableFactoryDefinition
   ): Promise<WizardEmbeddable | ErrorEmbeddable | DisabledEmbeddable> {
     try {
       const savedWizard = await getSavedWizardLoader().get(savedObjectId);
-
       const editPath = `${EDIT_PATH}/${savedObjectId}`;
-
       const editUrl = getHttp().basePath.prepend(`/app/${PLUGIN_ID}${editPath}`);
-
       const isLabsEnabled = getUISettings().get<boolean>(VISUALIZE_ENABLE_LABS_SETTING);
 
       if (!isLabsEnabled) {