🐛 Fix the dataViews switch bug

elastic · Aug 11, 2022 · d317513 · d317513
1 parent 07e3136
commit d317513
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/x-pack/plugins/lens/public/state_management/lens_slice.ts b/x-pack/plugins/lens/public/state_management/lens_slice.ts
@@ -374,19 +374,20 @@ export const makeLensReducer = (storeDeps: LensStoreDeps) => {
                   .getTableSpec()
                   .map(({ columnId }) => columnId)
                   .filter((columnId) => !nextTable.has(columnId));
-                const nextVisState = (newState.visualization || state.visualization).state;
                 const activeVisualization = visualizationMap[state.visualization.activeId];
+                let nextVisState = (newState.visualization || state.visualization).state;
                 removed.forEach((columnId) => {
-                  newState.visualization = {
-                    ...state.visualization,
-                    state: activeVisualization.removeDimension({
-                      layerId,
-                      columnId,
-                      prevState: nextVisState,
-                      frame,
-                    }),
-                  };
+                  nextVisState = activeVisualization.removeDimension({
+                    layerId,
+                    columnId,
+                    prevState: nextVisState,
+                    frame,
+                  });
                 });
+                newState.visualization = {
+                  ...state.visualization,
+                  state: nextVisState,
+                };
               }
             }
           }