diff --git a/web/src/app/chat/message/Messages.tsx b/web/src/app/chat/message/Messages.tsx
index 74d32cf95..3d9941393 100644
--- a/web/src/app/chat/message/Messages.tsx
+++ b/web/src/app/chat/message/Messages.tsx
@@ -58,7 +58,6 @@ import { ValidSources } from "@/lib/types";
 import { Tooltip } from "@/components/tooltip/Tooltip";
 import { useMouseTracking } from "./hooks";
 import { InternetSearchIcon } from "@/components/InternetSearchIcon";
-import { getTitleFromDocument } from "@/lib/sources";
 
 const TOOLS_WITH_CUSTOM_HANDLING = [
   SEARCH_TOOL_NAME,
@@ -445,7 +444,8 @@ export const AIMessage = ({
                                   >
                                     <Citation link={doc.link} index={ind + 1} />
                                     <p className="shrink truncate ellipsis break-all ">
-                                      {getTitleFromDocument(doc)}
+                                      {doc.semantic_identifier ||
+                                        doc.document_id}
                                     </p>
                                     <div className="ml-auto flex-none">
                                       {doc.is_internet ? (
diff --git a/web/src/lib/sources.ts b/web/src/lib/sources.ts
index 89891ef78..34d2a9235 100644
--- a/web/src/lib/sources.ts
+++ b/web/src/lib/sources.ts
@@ -301,9 +301,3 @@ function stripTrailingSlash(str: string) {
   }
   return str;
 }
-
-export const getTitleFromDocument = (document: DanswerDocument) => {
-  return stripTrailingSlash(document.document_id).split("/")[
-    stripTrailingSlash(document.document_id).split("/").length - 1
-  ];
-};