diff --git a/dotnet/test/E2E/SessionE2ETests.cs b/dotnet/test/E2E/SessionE2ETests.cs index 17658d78f..e47b6b21b 100644 --- a/dotnet/test/E2E/SessionE2ETests.cs +++ b/dotnet/test/E2E/SessionE2ETests.cs @@ -245,8 +245,7 @@ public async Task Should_Resume_A_Session_Using_A_New_Client() var session1 = await CreateSessionAsync(); var sessionId = session1.SessionId; - await session1.SendAsync(new MessageOptions { Prompt = "What is 1+1?" }); - var answer = await TestHelper.GetFinalAssistantMessageAsync(session1); + var answer = await session1.SendAndWaitAsync(new MessageOptions { Prompt = "What is 1+1?" }); Assert.NotNull(answer); Assert.Contains("2", answer!.Data.Content ?? string.Empty);