diff --git a/x-pack/solutions/security/plugins/security_solution/public/common/components/visualization_actions/use_save_to_library.tsx b/x-pack/solutions/security/plugins/security_solution/public/common/components/visualization_actions/use_save_to_library.tsx
index ab3ff52eab674..e9d7ad1c031d0 100644
--- a/x-pack/solutions/security/plugins/security_solution/public/common/components/visualization_actions/use_save_to_library.tsx
+++ b/x-pack/solutions/security/plugins/security_solution/public/common/components/visualization_actions/use_save_to_library.tsx
@@ -7,8 +7,8 @@
 
 import React, { useCallback, useMemo } from 'react';
 import { toMountPoint } from '@kbn/react-kibana-mount';
-import type { LensEmbeddableInput } from '@kbn/lens-plugin/public';
 import { unmountComponentAtNode } from 'react-dom';
+import type { SaveModalContainerProps } from '@kbn/lens-plugin/public/async_services';
 import { useKibana } from '../../lib/kibana';
 import type { LensAttributes } from './types';
 import { useRedirectToDashboardFromLens } from './use_redirect_to_dashboard_from_lens';
@@ -32,7 +32,7 @@ export const useSaveToLibrary = ({
     const targetDomElement = document.createElement('div');
     const mount = toMountPoint(
       <SaveModalComponent
-        initialInput={attributes as unknown as LensEmbeddableInput}
+        initialInput={{ attributes } as SaveModalContainerProps['initialInput']}
         onSave={() => unmountComponentAtNode(targetDomElement)}
         onClose={() => unmountComponentAtNode(targetDomElement)}
         originatingApp={APP_UI_ID}