diff --git a/web/app/components/datasets/create/step-two/index.tsx b/web/app/components/datasets/create/step-two/index.tsx
index 01cf933f30..5a3d8536dc 100644
--- a/web/app/components/datasets/create/step-two/index.tsx
+++ b/web/app/components/datasets/create/step-two/index.tsx
@@ -993,7 +993,7 @@ const StepTwo = ({
           >
             <div className='flex items-center gap-2'>
               <PreviewDocumentPicker
-                files={files.map(file => ({ name: file.name!, id: file.id!, extension: 'pdf' }))}
+                files={files as Array<Required<CustomFile>>}
                 onChange={(selected) => {
                   currentEstimateMutation.reset()
                   setPreviewFile(selected)