diff --git a/web/app/components/datasets/common/document-picker/preview-document-picker.tsx b/web/app/components/datasets/common/document-picker/preview-document-picker.tsx
index 856e90a28d..6d61cc4c69 100644
--- a/web/app/components/datasets/common/document-picker/preview-document-picker.tsx
+++ b/web/app/components/datasets/common/document-picker/preview-document-picker.tsx
@@ -54,7 +54,7 @@ const PreviewDocumentPicker: FC<Props> = ({
           <FileIcon name={name} extension={extension} size='md' />
           <div className='flex flex-col items-start ml-1'>
             <div className='flex items-center space-x-0.5'>
-              <span className={cn('system-md-semibold')}> {name || '--'}</span>
+              <span className={cn('system-md-semibold max-w-[200px] overflow-clip')}> {name || '--'}</span>
               <ArrowIcon className={'h-[18px] w-[18px] text-text-primary'} />
             </div>
           </div>
diff --git a/web/app/components/datasets/create/step-two/index.tsx b/web/app/components/datasets/create/step-two/index.tsx
index f006160ef0..ed8ee9064d 100644
--- a/web/app/components/datasets/create/step-two/index.tsx
+++ b/web/app/components/datasets/create/step-two/index.tsx
@@ -584,14 +584,16 @@ const StepTwo = ({
           <div className={s.label}>{t('datasetCreation.stepTwo.segmentation')}</div>
           <div className='max-w-[640px]'>
             <div className='space-y-4'>
-              {(!datasetId || [ChuckingMode.text, ChuckingMode.qa].includes(docForm))
+              {(!datasetId || [ChuckingMode.text, ChuckingMode.qa].includes(currentDataset!.doc_form))
                 && <OptionCard
                   title={t('datasetCreation.stepTwo.general')}
                   icon={<Image src={SettingCog} alt={t('datasetCreation.stepTwo.general')} />}
                   activeHeaderClassName='bg-gradient-to-r from-[#EFF0F9] to-[#F9FAFB]'
                   description={t('datasetCreation.stepTwo.generalTip')}
                   isActive={
-                    [ChuckingMode.text, ChuckingMode.qa].includes(docForm)
+                    [ChuckingMode.text, ChuckingMode.qa].includes(
+                      datasetId ? currentDataset!.doc_form : docForm,
+                    )
                   }
                   onSwitched={() =>
                     handleChangeDocform(ChuckingMode.text)
@@ -685,14 +687,16 @@ const StepTwo = ({
                   </div>
                 </OptionCard>}
               {
-                (!datasetId || docForm === ChuckingMode.parentChild)
+                (!datasetId || currentDataset!.doc_form === ChuckingMode.parentChild)
                 && <OptionCard
                   title={t('datasetCreation.stepTwo.parentChild')}
                   icon={<Image src={FamilyMod} alt={t('datasetCreation.stepTwo.parentChild')} />}
                   effectImg={OrangeEffect.src}
                   activeHeaderClassName='bg-gradient-to-r from-[#F9F1EE] to-[#F9FAFB]'
                   description={t('datasetCreation.stepTwo.parentChildTip')}
-                  isActive={docForm === ChuckingMode.parentChild}
+                  isActive={
+                    datasetId ? currentDataset!.doc_form === ChuckingMode.parentChild : docForm === ChuckingMode.parentChild
+                  }
                   onSwitched={() => handleChangeDocform(ChuckingMode.parentChild)}
                   actions={
                     <>