diff --git a/tests/ModelContextProtocol.Tests/Server/AutomaticInputRequiredStatusTests.cs b/tests/ModelContextProtocol.Tests/Server/AutomaticInputRequiredStatusTests.cs
index 788e56c50..1f5c51c6c 100644
--- a/tests/ModelContextProtocol.Tests/Server/AutomaticInputRequiredStatusTests.cs
+++ b/tests/ModelContextProtocol.Tests/Server/AutomaticInputRequiredStatusTests.cs
@@ -119,7 +119,7 @@ public async Task TaskStatus_TransitionsToInputRequired_DuringSampleAsync()
             finalStatus = await taskStore.GetTaskAsync(mcpTask.TaskId, cancellationToken: TestContext.Current.CancellationToken);
             maxAttempts--;
         }
-        while (finalStatus?.Status == McpTaskStatus.Working && maxAttempts > 0);
+        while (finalStatus?.Status is not McpTaskStatus.Completed && maxAttempts > 0);
 
         // Assert - Status should have been InputRequired during sampling
         Assert.Contains(McpTaskStatus.InputRequired, statusesDuringSampling);
@@ -218,7 +218,7 @@ public async Task TaskStatus_TransitionsToInputRequired_DuringElicitAsync()
             finalStatus = await taskStore.GetTaskAsync(mcpTask.TaskId, cancellationToken: TestContext.Current.CancellationToken);
             maxAttempts--;
         }
-        while (finalStatus?.Status == McpTaskStatus.Working && maxAttempts > 0);
+        while (finalStatus?.Status is not McpTaskStatus.Completed && maxAttempts > 0);
 
         // Assert - Status should have been InputRequired during elicitation
         Assert.Contains(McpTaskStatus.InputRequired, statusesDuringElicitation);