diff --git a/x-pack/legacy/plugins/index_management/public/components/mappings_editor/lib/utils.test.ts b/x-pack/legacy/plugins/index_management/public/components/mappings_editor/lib/utils.test.ts
index 4aa8aaf7c71a7..0431ea472643b 100644
--- a/x-pack/legacy/plugins/index_management/public/components/mappings_editor/lib/utils.test.ts
+++ b/x-pack/legacy/plugins/index_management/public/components/mappings_editor/lib/utils.test.ts
@@ -4,7 +4,7 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
-jest.mock('../constants', () => ({ DATA_TYPE_DEFINITION: {} }));
+jest.mock('../constants', () => ({ MAIN_DATA_TYPE_DEFINITION: {} }));
 
 import { isStateValid } from './utils';
 
diff --git a/x-pack/legacy/plugins/index_management/public/components/mappings_editor/lib/utils.ts b/x-pack/legacy/plugins/index_management/public/components/mappings_editor/lib/utils.ts
index 0896a7293f40c..3643ea081aef6 100644
--- a/x-pack/legacy/plugins/index_management/public/components/mappings_editor/lib/utils.ts
+++ b/x-pack/legacy/plugins/index_management/public/components/mappings_editor/lib/utils.ts
@@ -14,12 +14,11 @@ import {
   MainType,
   SubType,
   ChildFieldName,
-  TreeItem,
 } from '../types';
 
 import { MAIN_DATA_TYPE_DEFINITION, MAX_DEPTH_DEFAULT_EDITOR } from '../constants';
-
 import { State } from '../reducer';
+import { TreeItem } from '../components/tree';
 
 export const getUniqueId = () => {
   const dateNow = Date.now();