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);
     }
 }