diff --git a/test/Microsoft.AspNetCore.Http.Connections.Tests/TestWebSocketConnectionFeature.cs b/test/Microsoft.AspNetCore.Http.Connections.Tests/TestWebSocketConnectionFeature.cs
index c0711f4ace..f67dd94003 100644
--- a/test/Microsoft.AspNetCore.Http.Connections.Tests/TestWebSocketConnectionFeature.cs
+++ b/test/Microsoft.AspNetCore.Http.Connections.Tests/TestWebSocketConnectionFeature.cs
@@ -118,7 +118,7 @@ public override async Task<WebSocketReceiveResult> ReceiveAsync(ArraySegment<byt
                 {
                     if (_internalBuffer.Buffer == null || _internalBuffer.Buffer.Length == 0)
                     {
-                        await _input.WaitToReadAsync();
+                        await _input.WaitToReadAsync(cancellationToken);
 
                         if (_input.TryRead(out var message))
                         {