diff --git a/understand-anything-plugin/packages/dashboard/src/App.tsx b/understand-anything-plugin/packages/dashboard/src/App.tsx
index 9aa7edd..4aed9bf 100644
--- a/understand-anything-plugin/packages/dashboard/src/App.tsx
+++ b/understand-anything-plugin/packages/dashboard/src/App.tsx
@@ -448,7 +448,7 @@ function Dashboard({ accessToken }: { accessToken: string }) {
           <div className="flex items-center gap-4 w-max">
             <DiffToggle />
             {/* Detail level: file view (architecture) / class view (code structure) */}
-            {!isKnowledgeGraph && (
+            {!isKnowledgeGraph && viewMode !== "domain" && (
               <>
                 <div className="w-px h-5 bg-border-subtle" />
                 <div className="flex items-center bg-elevated rounded-lg p-0.5">
diff --git a/understand-anything-plugin/packages/dashboard/src/store.ts b/understand-anything-plugin/packages/dashboard/src/store.ts
index aff4339..b3b2a96 100644
--- a/understand-anything-plugin/packages/dashboard/src/store.ts
+++ b/understand-anything-plugin/packages/dashboard/src/store.ts
@@ -344,6 +344,8 @@ export const useDashboardStore = create<DashboardStore>()((set, get) => ({
     set({
       detailLevel: level,
       // Detail level changes which nodes are visible; cached positions stale.
+      // Reset fn toggle so it doesn't resurrect when re-entering class view.
+      showFunctionsInClassView: false,
       containerLayoutCache: new Map(),
       containerSizeMemory: new Map(),
       expandedContainers: new Set(),