diff --git a/web/app/components/datasets/common/retrieval-method-info/index.spec.tsx b/web/app/components/datasets/common/retrieval-method-info/index.spec.tsx
index fe2d623aeb..05750711dc 100644
--- a/web/app/components/datasets/common/retrieval-method-info/index.spec.tsx
+++ b/web/app/components/datasets/common/retrieval-method-info/index.spec.tsx
@@ -1,3 +1,4 @@
+import type { ReactNode } from 'react'
import { render, screen } from '@testing-library/react'
import { RETRIEVE_METHOD } from '@/types/app'
import { retrievalIcon } from '../../create/icons'
@@ -5,14 +6,14 @@ import RetrievalMethodInfo, { getIcon } from './index'
// Mock next/image
vi.mock('next/image', () => ({
- default: ({ src, alt, className }: any) => (
-
+ default: ({ src, alt, className }: { src: string, alt: string, className?: string }) => (
+
),
}))
// Mock RadioCard
vi.mock('@/app/components/base/radio-card', () => ({
- default: ({ title, description, chosenConfig, icon }: any) => (
+ default: ({ title, description, chosenConfig, icon }: { title: string, description: string, chosenConfig: ReactNode, icon: ReactNode }) => (