diff --git a/packages/ui/src/views/chatmessage/ChatMessage.js b/packages/ui/src/views/chatmessage/ChatMessage.js
index 98eef72a..506f02da 100644
--- a/packages/ui/src/views/chatmessage/ChatMessage.js
+++ b/packages/ui/src/views/chatmessage/ChatMessage.js
@@ -346,7 +346,9 @@ export const ChatMessage = ({ open, chatflowid, isDialog }) => {
                                                                 key={index}
                                                                 label={
                                                                     URL
-                                                                        ? `${URL.pathname.substring(0, 15)}...`
+                                                                        ? URL.pathname.substring(0, 15) === '/'
+                                                                            ? URL.host
+                                                                            : `${URL.pathname.substring(0, 15)}...`
                                                                         : `${source.pageContent.substring(0, 15)}...`
                                                                 }
                                                                 component='a'