diff --git a/samples/WebSocketPushServer/PushSession.cs b/samples/WebSocketPushServer/PushSession.cs
index 400778da9..115215d67 100644
--- a/samples/WebSocketPushServer/PushSession.cs
+++ b/samples/WebSocketPushServer/PushSession.cs
@@ -31,7 +31,7 @@ public int MessageClientReceived
 
         protected override async ValueTask OnSessionConnectedAsync()
         {
-            await this.SendAsync(this.SessionID);
+            await this.SendAsync(this.SessionID, CancellationToken.None);
         }
 
         public void Ack()
@@ -39,9 +39,9 @@ public void Ack()
             Interlocked.Increment(ref _messageClientReceived);
         }
 
-        public override async ValueTask SendAsync(string message)
+        public override async ValueTask SendAsync(string message, CancellationToken cancellationToken)
         {
-            await base.SendAsync(message);
+            await base.SendAsync(message, cancellationToken);
             Interlocked.Increment(ref _messageSent);
         }
 
diff --git a/samples/WebSocketPushServer/ServerPushMiddleware.cs b/samples/WebSocketPushServer/ServerPushMiddleware.cs
index 591b08cba..57bbb0c7d 100644
--- a/samples/WebSocketPushServer/ServerPushMiddleware.cs
+++ b/samples/WebSocketPushServer/ServerPushMiddleware.cs
@@ -59,7 +59,7 @@ private async ValueTask<int> Push()
 
             foreach (var s in _sessionContainer.GetSessions<PushSession>())
             {
-                await s.SendAsync(line);
+                await s.SendAsync(line, CancellationToken.None);
                 count++;
 
                 if (_stopped)