elastic
diff --git a/‎x-pack/plugins/lens/public/indexpattern_datasource/__mocks__/state_helpers.ts‎
Lines changed: 1 addition & 0 deletions b/‎x-pack/plugins/lens/public/indexpattern_datasource/__mocks__/state_helpers.ts‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎x-pack/plugins/lens/public/indexpattern_datasource/dimension_panel/dimension_editor.tsx‎
Lines changed: 10 additions & 10 deletions b/‎x-pack/plugins/lens/public/indexpattern_datasource/dimension_panel/dimension_editor.tsx‎
Lines changed: 10 additions & 10 deletions
@@ -17,4 +17,5 @@ export const {
   sortByField,
   hasField,
   updateLayerIndexPattern,
+  mergeLayer,
 } = actual;
@@ -16,15 +16,16 @@ import {
   EuiListGroupItemProps,
   EuiFormLabel,
 } from '@elastic/eui';
-import { IndexPatternDimensionEditorProps, OperationSupportMatrix } from './dimension_panel';
+import { IndexPatternDimensionEditorProps } from './dimension_panel';
+import { OperationSupportMatrix } from './operation_support';
 import { IndexPatternColumn, OperationType } from '../indexpattern';
 import {
   operationDefinitionMap,
   getOperationDisplay,
   buildColumn,
   changeField,
 } from '../operations';
-import { deleteColumn, changeColumn, updateColumnParam } from '../state_helpers';
+import { deleteColumn, changeColumn, updateColumnParam, mergeLayer } from '../state_helpers';
 import { FieldSelect } from './field_select';
 import { hasField, fieldIsInvalid } from '../utils';
 import { BucketNestingEditor } from './bucket_nesting_editor';
@@ -394,12 +395,11 @@ export function DimensionEditor(props: DimensionEditorProps) {
             <LabelInput
               value={selectedColumn.label}
               onChange={(value) => {
-                setState({
-                  ...state,
-                  layers: {
-                    ...state.layers,
-                    [layerId]: {
-                      ...state.layers[layerId],
+                setState(
+                  mergeLayer({
+                    state,
+                    layerId,
+                    newLayer: {
                       columns: {
                         ...state.layers[layerId].columns,
                         [columnId]: {
@@ -409,8 +409,8 @@ export function DimensionEditor(props: DimensionEditorProps) {
                         },
                       },
                     },
-                  },
-                });
+                  })
+                );
               }}
             />
           )}