diff --git a/packages/core/src/v3/schemas/messages.ts b/packages/core/src/v3/schemas/messages.ts
index 149b2c4c6..dc3bf6afe 100644
--- a/packages/core/src/v3/schemas/messages.ts
+++ b/packages/core/src/v3/schemas/messages.ts
@@ -1,5 +1,6 @@
 import { z } from "zod";
 import { TaskRunExecution, TaskRunExecutionResult } from "./common";
+import { WaitReason } from "./schemas";
 
 export const EnvironmentType = z.enum(["PRODUCTION", "STAGING", "DEVELOPMENT", "PREVIEW"]);
 export type EnvironmentType = z.infer<typeof EnvironmentType>;
@@ -356,7 +357,11 @@ export const ProdChildToWorkerMessages = {
   },
   CANCEL_CHECKPOINT: {
     message: z.object({
-      version: z.literal("v1").default("v1"),
+      version: z.enum(["v1", "v2"]).default("v2"),
+    }),
+    callback: z.object({
+      checkpointCanceled: z.boolean(),
+      reason: WaitReason.optional(),
     }),
   },
   WAIT_FOR_DURATION: {
diff --git a/packages/core/src/v3/schemas/schemas.ts b/packages/core/src/v3/schemas/schemas.ts
index f4d401aad..5c4fd6c12 100644
--- a/packages/core/src/v3/schemas/schemas.ts
+++ b/packages/core/src/v3/schemas/schemas.ts
@@ -396,7 +396,11 @@ export const ProdWorkerToCoordinatorMessages = {
   },
   CANCEL_CHECKPOINT: {
     message: z.object({
-      version: z.literal("v1").default("v1"),
+      version: z.enum(["v1", "v2"]).default("v2"),
+    }),
+    callback: z.object({
+      checkpointCanceled: z.boolean(),
+      reason: WaitReason.optional(),
     }),
   },
   TASK_HEARTBEAT: {