diff --git a/Samples~/WebsocketDemo/WebSocketBindings.cs b/Samples~/WebsocketDemo/WebSocketBindings.cs
index a78d236..7d311ad 100644
--- a/Samples~/WebsocketDemo/WebSocketBindings.cs
+++ b/Samples~/WebsocketDemo/WebSocketBindings.cs
@@ -3,6 +3,7 @@
 using System;
 using System.Collections.Generic;
 using System.Text;
+using System.Threading;
 using UnityEngine;
 using UnityEngine.UIElements;
 
@@ -44,6 +45,11 @@ public class WebSocketBindings : MonoBehaviour
         private WebSocket webSocket;
 
         private readonly List<Tuple<LogType, string>> logs = new();
+#if !UNITY_2022_3_OR_NEWER
+        // ReSharper disable once InconsistentNaming
+        private CancellationToken destroyCancellationToken => destroyCancellationTokenSource.Token;
+        private CancellationTokenSource destroyCancellationTokenSource = new CancellationTokenSource();
+#endif
 
         private void OnValidate()
         {
@@ -58,7 +64,6 @@ private void Awake()
             OnValidate();
 
             var root = uiDocument.rootVisualElement;
-
             statusLabel = root.Q<Label>("status-label");
             fpsLabel = root.Q<Label>("fps-label");
             addressTextField = root.Q<TextField>("address-text-field");
@@ -147,6 +152,15 @@ private void OnDisable()
             clearLogsButton.clicked -= OnClearLogsClick;
         }
 
+        private void OnDestroy()
+        {
+#if !UNITY_2022_3_OR_NEWER
+            destroyCancellationTokenSource?.Cancel();
+            destroyCancellationTokenSource?.Dispose();
+            destroyCancellationTokenSource = null;
+#endif
+        }
+
         private async void OnConnectClick()
         {
             if (webSocket != null)
@@ -181,7 +195,7 @@ private async void OnDisconnectClick()
 
             try
             {
-                await webSocket.CloseAsync();
+                await webSocket.CloseAsync(cancellationToken: CancellationToken.None);
             }
             catch (Exception e)
             {
diff --git a/package.json b/package.json
index ea346c4..04408ad 100644
--- a/package.json
+++ b/package.json
@@ -3,7 +3,7 @@
   "displayName": "Utilities.WebSockets",
   "description": "A simple websocket package for Unity (UPM)",
   "keywords": [],
-  "version": "1.0.0-preview.1",
+  "version": "1.0.0-preview.2",
   "unity": "2021.3",
   "documentationUrl": "https://github.com/RageAgainstThePixel/com.utilities.websockets#documentation",
   "changelogUrl": "https://github.com/RageAgainstThePixel/com.utilities.websockets/releases",
@@ -17,7 +17,7 @@
     "url": "https://github.com/StephenHodgson"
   },
   "dependencies": {
-    "com.utilities.async": "2.1.5"
+    "com.utilities.async": "2.1.7"
   },
   "samples": [
     {