diff --git a/src/legacy/ui/public/visualize/loader/pipeline_data_loader.ts b/src/legacy/ui/public/visualize/loader/pipeline_data_loader.ts
index 301e7f1a7e378..e9ccaf75e63f9 100644
--- a/src/legacy/ui/public/visualize/loader/pipeline_data_loader.ts
+++ b/src/legacy/ui/public/visualize/loader/pipeline_data_loader.ts
@@ -31,6 +31,7 @@ export class PipelineDataLoader {
       {},
       {
         getInitialContext: () => ({
+          type: 'kibana_context',
           query: params.query,
           timeRange: params.timeRange,
           filters: params.filters
diff --git a/src/legacy/ui/public/visualize/loader/pipeline_helpers/run_pipeline.ts b/src/legacy/ui/public/visualize/loader/pipeline_helpers/run_pipeline.ts
index 6c51d5059d011..1551ab755ea5b 100644
--- a/src/legacy/ui/public/visualize/loader/pipeline_helpers/run_pipeline.ts
+++ b/src/legacy/ui/public/visualize/loader/pipeline_helpers/run_pipeline.ts
@@ -21,10 +21,9 @@
 import { fromExpression } from '@kbn/interpreter/common';
 // @ts-ignore
 import { getInterpreter } from 'plugins/interpreter/interpreter';
-// @ts-ignore
-import { KibanaContext } from 'plugins/interpreter/types';
 
 import { Adapters } from 'ui/inspector';
+import { KibanaContext } from '../../../../../core_plugins/interpreter/public';
 
 type getInitialContextFunction = () => KibanaContext;