diff --git a/src/remote/optimization.ts b/src/remote/optimization.ts
index 48186ec..f2d7d46 100644
--- a/src/remote/optimization.ts
+++ b/src/remote/optimization.ts
@@ -29,6 +29,7 @@ export const LiveQueryOptimization = z.discriminatedUnion("state", [
     indexRecommendations: z.array(IndexRecommendation),
     indexesUsed: z.array(z.string()),
     explainPlan: z.custom<PostgresExplainStage>(),
+    optimizedExplainPlan: z.custom<PostgresExplainStage>(),
   }),
   z.object({
     state: z.literal("no_improvement_found"),
diff --git a/src/remote/query-optimizer.ts b/src/remote/query-optimizer.ts
index 35f6b53..f1bf79b 100644
--- a/src/remote/query-optimizer.ts
+++ b/src/remote/query-optimizer.ts
@@ -340,6 +340,7 @@ export class QueryOptimizer extends EventEmitter<EventMap> {
             indexRecommendations,
             indexesUsed,
             explainPlan,
+            optimizedExplainPlan: result.explainPlan,
           };
         }
       }
@@ -422,6 +423,7 @@ export class QueryOptimizer extends EventEmitter<EventMap> {
       indexRecommendations,
       indexesUsed,
       explainPlan,
+      optimizedExplainPlan: result.explainPlan,
     };
   }