diff --git a/surfsense_web/app/dashboard/[workspace_id]/onboard/page.tsx b/surfsense_web/app/dashboard/[workspace_id]/onboard/page.tsx
index 861ceaa07..9da14ef4a 100644
--- a/surfsense_web/app/dashboard/[workspace_id]/onboard/page.tsx
+++ b/surfsense_web/app/dashboard/[workspace_id]/onboard/page.tsx
@@ -74,7 +74,7 @@ export default function OnboardPage() {
: ;
diff --git a/surfsense_web/components/assistant-ui/thread.tsx b/surfsense_web/components/assistant-ui/thread.tsx
index 3716c59aa..34ff4888f 100644
--- a/surfsense_web/components/assistant-ui/thread.tsx
+++ b/surfsense_web/components/assistant-ui/thread.tsx
@@ -108,6 +108,7 @@ import { useElectronAPI } from "@/hooks/use-platform";
import { captureDisplayToPngDataUrl } from "@/lib/chat/display-media-capture";
import { getMentionDocKey } from "@/lib/chat/mention-doc-key";
import { slideoutOpenedTickAtom } from "@/lib/layout-events";
+import { getWorkspaceIdNumber } from "@/lib/route-params";
import { cn } from "@/lib/utils";
import {
DocumentMentionPicker,
@@ -458,7 +459,9 @@ const Composer: FC = () => {
const prevMentionedDocsRef = useRef