diff --git a/x-pack/plugins/maps/public/classes/layers/choropleth_layer_wizard/layer_template.test.tsx b/x-pack/plugins/maps/public/classes/layers/choropleth_layer_wizard/layer_template.test.tsx
index e9d25320712036..ecb86756e1ca73 100644
--- a/x-pack/plugins/maps/public/classes/layers/choropleth_layer_wizard/layer_template.test.tsx
+++ b/x-pack/plugins/maps/public/classes/layers/choropleth_layer_wizard/layer_template.test.tsx
@@ -22,11 +22,12 @@ import { BOUNDARIES_SOURCE, LayerTemplate } from './layer_template';
 const renderWizardArguments = {
   previewLayers: () => {},
   mapColors: [],
-  isIndexingTriggered: false,
-  onRemove: () => {},
-  onIndexReady: () => {},
-  importSuccessHandler: () => {},
-  importErrorHandler: () => {},
+  currentStepId: null,
+  enableNextBtn: () => {},
+  disableNextBtn: () => {},
+  startStepLoading: () => {},
+  stopStepLoading: () => {},
+  advanceToNextStep: () => {},
 };
 
 test('should render elasticsearch UI when left source is BOUNDARIES_SOURCE.ELASTICSEARCH', async () => {