diff --git a/src/legacy/ui/public/saved_objects/types.ts b/src/legacy/ui/public/saved_objects/types.ts
index bccf73917882a..81981da45604d 100644
--- a/src/legacy/ui/public/saved_objects/types.ts
+++ b/src/legacy/ui/public/saved_objects/types.ts
@@ -17,7 +17,7 @@
  * under the License.
  */
 import { ChromeStart, OverlayStart, SavedObjectsClientContract } from 'kibana/public';
-import { SearchSource, SearchSourceContract } from 'ui/courier';
+import { ISearchSource } from 'ui/courier';
 import { SavedObjectAttributes, SavedObjectReference } from 'kibana/server';
 import { IndexPatternsContract } from '../../../../plugins/data/public';
 import { IndexPattern } from '../../../core_plugins/data/public';
@@ -42,7 +42,7 @@ export interface SavedObject {
   lastSavedTitle: string;
   migrationVersion?: Record<string, any>;
   save: (saveOptions: SavedObjectSaveOpts) => Promise<string>;
-  searchSource?: SearchSourceContract;
+  searchSource?: ISearchSource;
   showInRecentlyAccessed: boolean;
   title: string;
 }
@@ -83,7 +83,7 @@ export interface SavedObjectConfig {
   mapping?: any;
   migrationVersion?: Record<string, any>;
   path?: string;
-  searchSource?: SearchSource | boolean;
+  searchSource?: ISearchSource | boolean;
   type?: string;
 }
 
diff --git a/x-pack/legacy/plugins/lens/public/editor_frame_plugin/merge_tables.test.ts b/x-pack/legacy/plugins/lens/public/editor_frame_plugin/merge_tables.test.ts
index 8f124c25542c7..ffb8be1deaa9e 100644
--- a/x-pack/legacy/plugins/lens/public/editor_frame_plugin/merge_tables.test.ts
+++ b/x-pack/legacy/plugins/lens/public/editor_frame_plugin/merge_tables.test.ts
@@ -8,6 +8,8 @@ import moment from 'moment';
 import { mergeTables } from './merge_tables';
 import { KibanaDatatable } from 'src/plugins/expressions/public';
 
+jest.mock('ui/new_platform');
+
 describe('lens_merge_tables', () => {
   it('should produce a row with the nested table as defined', () => {
     const sampleTable1: KibanaDatatable = {