From d293627e083dea1bea73f09219f0c08cbb4db89f Mon Sep 17 00:00:00 2001
From: Loi Nguyen <loinguyen@local>
Date: Mon, 20 Jul 2026 03:36:18 +0700
Subject: [PATCH] Stabilize task cancellation cleanup test

---
 .../Server/TaskCancellationIntegrationTests.cs                   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/ModelContextProtocol.Tests/Server/TaskCancellationIntegrationTests.cs b/tests/ModelContextProtocol.Tests/Server/TaskCancellationIntegrationTests.cs
index e751bdcdc..941a75734 100644
--- a/tests/ModelContextProtocol.Tests/Server/TaskCancellationIntegrationTests.cs
+++ b/tests/ModelContextProtocol.Tests/Server/TaskCancellationIntegrationTests.cs
@@ -106,6 +106,7 @@ public async Task TaskTool_CancellationToken_GetTaskShowsWorkingBeforeCancel()
 
         // Cleanup
         await client.CancelTaskAsync(taskId, ct);
+        await _toolCancellationFired.Task.WaitAsync(TestConstants.DefaultTimeout, ct);
     }
 }