diff --git a/packages/ui/src/views/docstore/DeleteDocStoreDialog.jsx b/packages/ui/src/views/docstore/DeleteDocStoreDialog.jsx
index d1e67c940..072d0258c 100644
--- a/packages/ui/src/views/docstore/DeleteDocStoreDialog.jsx
+++ b/packages/ui/src/views/docstore/DeleteDocStoreDialog.jsx
@@ -226,7 +226,7 @@ const DeleteDocStoreDialog = ({ show, dialogProps, onCancel, onDelete }) => {
                 <Button onClick={onCancel} color='primary'>
                     Cancel
                 </Button>
-                <Button variant='contained' onClick={() => onDelete(dialogProps.type, removeFromVS)} color='error'>
+                <Button variant='contained' onClick={() => onDelete(dialogProps.type, dialogProps.file, removeFromVS)} color='error'>
                     Delete
                 </Button>
             </DialogActions>
diff --git a/packages/ui/src/views/docstore/DocumentStoreDetail.jsx b/packages/ui/src/views/docstore/DocumentStoreDetail.jsx
index 150fc53fe..a1f1b7821 100644
--- a/packages/ui/src/views/docstore/DocumentStoreDetail.jsx
+++ b/packages/ui/src/views/docstore/DocumentStoreDetail.jsx
@@ -184,8 +184,9 @@ const DocumentStoreDetails = () => {
         }
     }
 
-    const onDocStoreDelete = async (type, removeFromVectorStore) => {
+    const onDocStoreDelete = async (type, file, removeFromVectorStore) => {
         setBackdropLoading(true)
+        setShowDeleteDocStoreDialog(false)
         if (type === 'STORE') {
             if (removeFromVectorStore) {
                 await deleteVectorStoreDataFromStore(storeId)
@@ -272,7 +273,8 @@ const DocumentStoreDetails = () => {
             description: `Delete Loader ${file.loaderName} ? This will delete all the associated document chunks.`,
             vectorStoreConfig,
             recordManagerConfig,
-            type: 'LOADER'
+            type: 'LOADER',
+            file
         }
 
         setDeleteDocStoreDialogProps(props)