diff --git a/webui/src/web-components/sketch-app-shell.ts b/webui/src/web-components/sketch-app-shell.ts
index 967f502..29e12ba 100644
--- a/webui/src/web-components/sketch-app-shell.ts
+++ b/webui/src/web-components/sketch-app-shell.ts
@@ -1352,6 +1352,7 @@
                 .toolCalls=${this.containerState?.outstanding_tool_calls || []}
                 .firstMessageIndex=${this.containerState?.first_message_index ||
                 0}
+                .state=${this.containerState}
               ></sketch-timeline>
             </div>
           </div>