diff --git a/src/lib/components/chat/MessageInput.svelte b/src/lib/components/chat/MessageInput.svelte
index c8baab5d..f76a74d6 100644
--- a/src/lib/components/chat/MessageInput.svelte
+++ b/src/lib/components/chat/MessageInput.svelte
@@ -107,6 +107,7 @@
 
 		if (res) {
 			doc.upload_status = true;
+			doc.collection_name = res.collection_name;
 			files = files;
 		}
 	};