diff --git a/webui/src/web-components/sketch-timeline.ts b/webui/src/web-components/sketch-timeline.ts
index 3dd17a2..c450330 100644
--- a/webui/src/web-components/sketch-timeline.ts
+++ b/webui/src/web-components/sketch-timeline.ts
@@ -2,7 +2,7 @@
 import { PropertyValues } from "lit";
 import { repeat } from "lit/directives/repeat.js";
 import { customElement, property, state } from "lit/decorators.js";
-import { AgentMessage } from "../types";
+import { AgentMessage, State } from "../types";
 import "./sketch-timeline-message";
 import { Ref } from "lit/directives/ref";
 
@@ -31,6 +31,9 @@
   @property({ attribute: false })
   firstMessageIndex: number = 0;
 
+  @property({ attribute: false })
+  state: State | null = null;
+
   static styles = css`
     /* Hide views initially to prevent flash of content */
     .timeline-container .timeline,
@@ -402,6 +405,7 @@
                   .previousMessage=${previousMessage}
                   .open=${false}
                   .firstMessageIndex=${this.firstMessageIndex}
+                  .state=${this.state}
                 ></sketch-timeline-message>`;
               },
             )}