diff --git a/server/src/__tests__/heartbeat-comment-wake-batching.test.ts b/server/src/__tests__/heartbeat-comment-wake-batching.test.ts
index 9f087a58241..bdd997d4553 100644
--- a/server/src/__tests__/heartbeat-comment-wake-batching.test.ts
+++ b/server/src/__tests__/heartbeat-comment-wake-batching.test.ts
@@ -704,8 +704,29 @@ describeEmbeddedPostgres("heartbeat comment wake batching", () => {
       });
 
       expect(firstRun).not.toBeNull();
+      if (!firstRun?.wakeupRequestId) throw new Error("Expected first feedback run to retain its wake request");
       await waitFor(() => gateway.getAgentPayloads().length === 1);
 
+      await db.insert(issueComments).values({
+        companyId,
+        issueId,
+        authorAgentId: agentId,
+        createdByRunId: firstRun.id,
+        body: "Initial feedback handled before cancellation.",
+        metadata: {
+          version: 1,
+          feedbackDisposition: {
+            kind: "feedback_delivery",
+            rootWakeupRequestId: firstRun.wakeupRequestId,
+            handledCommentIds: [comment1.id],
+          },
+          sections: [{
+            title: "Feedback disposition",
+            rows: [{ type: "key_value", label: "Handled comment", value: comment1.id }],
+          }],
+        },
+      });
+
       const queuedComment = await db
         .insert(issueComments)
         .values({