diff --git a/package.json b/package.json
index 481d1ea..6baa5bf 100644
--- a/package.json
+++ b/package.json
@@ -4,7 +4,7 @@
   "description": "Transform Visualization",
   "main": "index.js",
   "kibana": {
-    "version": "7.7.1",
+    "version": "7.8.1",
     "templateVersion": "1.0.0"
   },
   "license": "Apache-2.0",
diff --git a/public/legacy.ts b/public/legacy.ts
index eec5316..26cd493 100644
--- a/public/legacy.ts
+++ b/public/legacy.ts
@@ -1,13 +1,12 @@
 import { PluginInitializerContext } from 'kibana/public';
 import { npSetup, npStart } from 'ui/new_platform';
 
-import { setup as visualizationsSetup } from '../../../src/legacy/core_plugins/visualizations/public/np_ready/public/legacy';
 import { TransformPluginSetupDependencies } from './plugin';
 import { plugin } from '.';
 
 const plugins: Readonly<TransformPluginSetupDependencies> = {
   expressions: npSetup.plugins.expressions,
-  visualizations: visualizationsSetup,
+  visualizations: npSetup.plugins.visualizations,
   data: npSetup.plugins.data,
   dataStart: npStart.plugins.data,
 };
diff --git a/public/transform_vis.ts b/public/transform_vis.ts
index 505e0b9..d8824ca 100644
--- a/public/transform_vis.ts
+++ b/public/transform_vis.ts
@@ -1,13 +1,13 @@
 import { i18n } from '@kbn/i18n';
 
 import { IUiSettingsClient } from 'kibana/public';
-import { DefaultEditorSize } from '../../../src/legacy/core_plugins/vis_default_editor/public';
+import { DefaultEditorSize } from '../../../src/plugins/vis_default_editor/public';
 
 import { getTransformVisWrapper } from './transform_vis_controller';
 import { getTransformOptions } from './transform_options';
 import { getTransformRequestHandler } from './request_handler';
 import { DataPublicPluginSetup } from '../../../src/plugins/data/public';
-import { LegacyApiCaller } from '../../../src/plugins/data/public/search/es_client';
+import { LegacyApiCaller } from '../../../src/plugins/data/public/search/legacy/es_client';
 
 export const createTransformVisDefinition = ({
   uiSettings,