From 3ac93daf4f78b58bbd1df390b64c2d179dd767f3 Mon Sep 17 00:00:00 2001
From: Oliver Booth <git@olivr.me>
Date: Tue, 13 Feb 2024 00:33:04 +0000
Subject: [PATCH] refactor: pass cancellation token to SetCanceled

---
 TcpDotNet/ClientNode.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/TcpDotNet/ClientNode.cs b/TcpDotNet/ClientNode.cs
index bc8ebd9..28ebda8 100644
--- a/TcpDotNet/ClientNode.cs
+++ b/TcpDotNet/ClientNode.cs
@@ -295,11 +295,11 @@ public abstract class ClientNode : Node
                 }
                 catch (TaskCanceledException)
                 {
-                    completionSource.SetCanceled();
+                    completionSource.SetCanceled(cancellationToken);
                 }
             }
 
-            completionSource.SetCanceled();
+            completionSource.SetCanceled(cancellationToken);
         }, cancellationToken);
 
         var packet = (TPacket)await Task.Run(() => completionSource.Task, cancellationToken);