diff --git a/packages/ui/src/views/assistants/custom/CustomAssistantConfigurePreview.jsx b/packages/ui/src/views/assistants/custom/CustomAssistantConfigurePreview.jsx
index e782ddb4..0c9e5ace 100644
--- a/packages/ui/src/views/assistants/custom/CustomAssistantConfigurePreview.jsx
+++ b/packages/ui/src/views/assistants/custom/CustomAssistantConfigurePreview.jsx
@@ -321,7 +321,7 @@ const CustomAssistantConfigurePreview = () => {
                 const docStoreOption = documentStoreOptions.find((ds) => ds.name === selectedDocumentStores[i].id)
                 // convert to small case and replace space with underscore
                 const name = (docStoreOption?.label || '').toLowerCase().replace(/ /g, '_')
-                const desc = docStoreOption?.description || ''
+                const desc = selectedDocumentStores[i].description || docStoreOption?.description || ''
 
                 set(retrieverToolNodeData, 'inputs', {
                     name,