1 files changed, 3 insertions, 3 deletions
diff --git a/src/SignalR/server/SignalR/test/HubConnectionHandlerTests.ClientResult.cs b/src/SignalR/server/SignalR/test/HubConnectionHandlerTests.ClientResult.cs
index 42c523e1eb..91133c9604 100644
--- a/src/SignalR/server/SignalR/test/HubConnectionHandlerTests.ClientResult.cs
+++ b/src/SignalR/server/SignalR/test/HubConnectionHandlerTests.ClientResult.cs
@@ -125,7 +125,7 @@ public partial class HubConnectionHandlerTests
 
                 // Hub asks client for a result, this is an invocation message with an ID
                 var closeMessage = Assert.IsType<CloseMessage>(await client.ReadAsync().DefaultTimeout());
-                Assert.Equal("Connection closed with an error. InvalidOperationException: Client results inside OnConnectedAsync or OnDisconnectedAsync Hub methods are not allowed.", closeMessage.Error);
+                Assert.Equal("Connection closed with an error. InvalidOperationException: Client results inside OnConnectedAsync Hub methods are not allowed.", closeMessage.Error);
             }
         }
 
@@ -156,8 +156,8 @@ public partial class HubConnectionHandlerTests
                 var closeMessage = Assert.IsType<CloseMessage>(await client.ReadAsync().DefaultTimeout());
                 Assert.Null(closeMessage.Error);
 
-                var ex = await Assert.ThrowsAsync<InvalidOperationException>(() => connectionHandlerTask).DefaultTimeout();
-                Assert.Equal("Client results inside OnConnectedAsync or OnDisconnectedAsync Hub methods are not allowed.", ex.Message);
+                var ex = await Assert.ThrowsAsync<IOException>(() => connectionHandlerTask).DefaultTimeout();
+                Assert.Equal($"Connection '{client.Connection.ConnectionId}' disconnected.", ex.Message);
             }
         }