diff --git a/gadget-code/frontend/src/lib/socket.ts b/gadget-code/frontend/src/lib/socket.ts
index f1eb4f9..0158d9e 100644
--- a/gadget-code/frontend/src/lib/socket.ts
+++ b/gadget-code/frontend/src/lib/socket.ts
@@ -1,6 +1,6 @@
 import { createContext } from "react";
 import { io, Socket } from "socket.io-client";
-import type { ChatSession } from "./api";
+import type { ChatSession, IWorkOrderCompleteStats } from "./api";
 
 /**
  * Web Worker for heartbeat timing — avoids browser tab throttling.
@@ -30,6 +30,7 @@ export interface ServerToClientEvents {
     turnId: string,
     success: boolean,
     message?: string,
+    stats?: IWorkOrderCompleteStats,
   ) => void;
   log: (
     timestamp: string,
@@ -90,6 +91,7 @@ export interface SocketEvents {
     turnId: string,
     success: boolean,
     message?: string,
+    stats?: IWorkOrderCompleteStats,
   ) => void;
   "agent:thinking": (data: { agentId: string; thinking: string }) => void;
   "agent:response": (data: { agentId: string; chunk: string }) => void;
@@ -228,8 +230,8 @@ class SocketClient {
 
     this.socket.on(
       "workOrderComplete",
-      (turnId: string, success: boolean, message?: string) => {
-        this.emit("workOrderComplete", turnId, success, message);
+      (turnId: string, success: boolean, message?: string, stats?: IWorkOrderCompleteStats) => {
+        this.emit("workOrderComplete", turnId, success, message, stats);
       },
     );