diff --git a/modules/ai/assets/js/editor/app.js b/modules/ai/assets/js/editor/app.js
index 726d3b032f52..88f21a060700 100644
--- a/modules/ai/assets/js/editor/app.js
+++ b/modules/ai/assets/js/editor/app.js
@@ -1,8 +1,8 @@
 import { ThemeProvider, DirectionProvider } from '@elementor/ui';
 import PropTypes from 'prop-types';
 import PageContent from './page-content';
-import { onConnect } from './helpers';
 import { RequestIdsProvider } from './context/requests-ids';
+import { onConnect } from './utils/editor-integration';
 
 const App = ( props ) => {
 	return (