diff --git a/web/app/components/datasets/create/step-two/hooks/use-segmentation-state.ts b/web/app/components/datasets/create/step-two/hooks/use-segmentation-state.ts
index 41c34a9a96..503704276e 100644
--- a/web/app/components/datasets/create/step-two/hooks/use-segmentation-state.ts
+++ b/web/app/components/datasets/create/step-two/hooks/use-segmentation-state.ts
@@ -62,8 +62,11 @@ export const useSegmentationState = (options: UseSegmentationStateOptions = {})
   const [summaryIndexSetting, setSummaryIndexSetting] = useState<SummaryIndexSettingType | undefined>(initialSummaryIndexSetting)
   const summaryIndexSettingRef = useRef<SummaryIndexSettingType | undefined>(initialSummaryIndexSetting)
   const handleSummaryIndexSettingChange = useCallback((payload: SummaryIndexSettingType) => {
-    setSummaryIndexSetting({ ...summaryIndexSettingRef.current, ...payload })
-    summaryIndexSettingRef.current = { ...summaryIndexSettingRef.current, ...payload }
+    setSummaryIndexSetting((prev) => {
+      const newSetting = { ...prev, ...payload }
+      summaryIndexSettingRef.current = newSetting
+      return newSetting
+    })
   }, [])
 
   // Parent-child config
diff --git a/web/app/components/datasets/settings/form/index.tsx b/web/app/components/datasets/settings/form/index.tsx
index a80efff95f..1993c9fd8d 100644
--- a/web/app/components/datasets/settings/form/index.tsx
+++ b/web/app/components/datasets/settings/form/index.tsx
@@ -78,10 +78,10 @@ const Form = () => {
         },
   )
   const [summaryIndexSetting, setSummaryIndexSetting] = useState(currentDataset?.summary_index_setting)
-  const summaryIndexSettingRef = useRef(currentDataset?.summary_index_setting)
   const handleSummaryIndexSettingChange = useCallback((payload: SummaryIndexSettingType) => {
-    setSummaryIndexSetting({ ...summaryIndexSettingRef.current, ...payload })
-    summaryIndexSettingRef.current = { ...summaryIndexSettingRef.current, ...payload }
+    setSummaryIndexSetting((prev) => {
+      return { ...prev, ...payload }
+    })
   }, [])
   const { data: rerankModelList } = useModelList(ModelTypeEnum.rerank)
   const { data: embeddingModelList } = useModelList(ModelTypeEnum.textEmbedding)