diff --git a/loop/webui/src/data.ts b/loop/webui/src/data.ts
index 9eea954..9b5aca9 100644
--- a/loop/webui/src/data.ts
+++ b/loop/webui/src/data.ts
@@ -1,4 +1,4 @@
-import { TimelineMessage } from "./types";
+import { AgentMessage } from "./types";
 import { formatNumber } from "./utils";
 
 /**
@@ -42,7 +42,7 @@
   private isPollingEnabled: boolean = true;
   private isFirstLoad: boolean = true;
   private connectionStatus: ConnectionStatus = "disabled";
-  private messages: TimelineMessage[] = [];
+  private messages: AgentMessage[] = [];
   private timelineState: TimelineState | null = null;
 
   // Event listeners
@@ -76,7 +76,7 @@
   /**
    * Get all messages
    */
-  public getMessages(): TimelineMessage[] {
+  public getMessages(): AgentMessage[] {
     return this.messages;
   }