diff --git a/src/Shared/runtime/Quic/Implementations/Mock/MockConnection.cs b/src/Shared/runtime/Quic/Implementations/Mock/MockConnection.cs
index 361474bcb1c8199500350fe2a89cfe437f65dd17..cba2f936ef8dd2ea00d3669c247ddc4c9de5430f 100644
--- a/src/Shared/runtime/Quic/Implementations/Mock/MockConnection.cs
+++ b/src/Shared/runtime/Quic/Implementations/Mock/MockConnection.cs
@@ -93,7 +93,7 @@ namespace System.Net.Quic.Implementations.Mock
             int bytesRead = 0;
             do
             {
-                bytesRead += await socket.ReceiveAsync(buffer.AsMemory().Slice(bytesRead), SocketFlags.None).ConfigureAwait(false);
+                bytesRead += await socket.ReceiveAsync(buffer.AsMemory().Slice(bytesRead), SocketFlags.None, cancellationToken).ConfigureAwait(false);
             } while (bytesRead != buffer.Length);
 
             int peerListenPort = BinaryPrimitives.ReadInt32LittleEndian(buffer);
@@ -163,7 +163,7 @@ namespace System.Net.Quic.Implementations.Mock
             int bytesRead = 0;
             do
             {
-                bytesRead += await socket.ReceiveAsync(buffer.AsMemory().Slice(bytesRead), SocketFlags.None).ConfigureAwait(false);
+                bytesRead += await socket.ReceiveAsync(buffer.AsMemory().Slice(bytesRead), SocketFlags.None, cancellationToken).ConfigureAwait(false);
             } while (bytesRead != buffer.Length);
 
             long streamId = BinaryPrimitives.ReadInt64LittleEndian(buffer);
diff --git a/src/Shared/runtime/Quic/Implementations/Mock/MockListener.cs b/src/Shared/runtime/Quic/Implementations/Mock/MockListener.cs
index f4c0cfdf2c2fdb4cdcc7fc6dc859fc3c6c0d6ca8..e7b24548d2cf6d1d171136d66e3743775a655812 100644
--- a/src/Shared/runtime/Quic/Implementations/Mock/MockListener.cs
+++ b/src/Shared/runtime/Quic/Implementations/Mock/MockListener.cs
@@ -45,7 +45,7 @@ namespace System.Net.Quic.Implementations.Mock
             int bytesRead = 0;
             do
             {
-                bytesRead += await socket.ReceiveAsync(buffer.AsMemory().Slice(bytesRead), SocketFlags.None).ConfigureAwait(false);
+                bytesRead += await socket.ReceiveAsync(buffer.AsMemory().Slice(bytesRead), SocketFlags.None, cancellationToken).ConfigureAwait(false);
             } while (bytesRead != buffer.Length);
 
             int peerListenPort = BinaryPrimitives.ReadInt32LittleEndian(buffer);