diff --git a/extensions/session-header.ts b/extensions/session-header.ts
index 5a410bd..0fe5f7a 100644
--- a/extensions/session-header.ts
+++ b/extensions/session-header.ts
@@ -122,7 +122,7 @@ export default function (pi: ExtensionAPI) {
               pi.setSessionName(sessionTitle);
               ctx.ui.setWidget(
                 "session-header",
-                makeWidget(sessionTitle, ctx.ui.theme),
+                (_ui: any, theme: any) => makeWidget(sessionTitle, theme),
               );
             }
           }).catch(() => {
@@ -136,7 +136,7 @@ export default function (pi: ExtensionAPI) {
       }
     }
 
-    ctx.ui.setWidget("session-header", makeWidget(sessionTitle, ctx.ui.theme));
+    ctx.ui.setWidget("session-header", (_ui: any, theme: any) => makeWidget(sessionTitle, theme));
   });
 
   // Erste Benutzereingabe: Intention erkennen statt 1:1 übernehmen
@@ -147,14 +147,14 @@ export default function (pi: ExtensionAPI) {
     // Interim: sofort gekürzten Originaltext anzeigen
     sessionTitle = truncate(event.text);
     pi.setSessionName(sessionTitle);
-    ctx.ui.setWidget("session-header", makeWidget(sessionTitle, ctx.ui.theme));
+    ctx.ui.setWidget("session-header", (_ui: any, theme: any) => makeWidget(sessionTitle, theme));
 
     // Async: LLM-Zusammenfassung nachladen
     generateIntentionSummary(event.text, ctx).then((summary) => {
       if (summary) {
         sessionTitle = summary;
         pi.setSessionName(sessionTitle);
-        ctx.ui.setWidget("session-header", makeWidget(sessionTitle, ctx.ui.theme));
+        ctx.ui.setWidget("session-header", (_ui: any, theme: any) => makeWidget(sessionTitle, theme));
       }
     }).catch(() => {
       // Silent fail — truncate reicht als Fallback
diff --git a/extensions/session-index.ts b/extensions/session-index.ts
index 86bbaf3..799d526 100644
--- a/extensions/session-index.ts
+++ b/extensions/session-index.ts
@@ -237,7 +237,7 @@ export default function (pi: ExtensionAPI) {
       writeIndex(sessions);
 
       // Widget anzeigen (oberhalb des Editors)
-      ctx.ui.setWidget("session-index", makeIndexWidget(sessions, ctx.ui.theme));
+      ctx.ui.setWidget("session-index", (_ui: any, theme: any) => makeIndexWidget(sessions, theme));
 
       // Prüfen ob Vorgänger-Session einen Handover hat
       if (sessions.length >= 2 && sessions[1].hasHandover) {