diff --git a/examples~/quickstart/client/Program.cs b/examples~/quickstart/client/Program.cs
index 8327d976..f6978ff6 100644
--- a/examples~/quickstart/client/Program.cs
+++ b/examples~/quickstart/client/Program.cs
@@ -144,12 +144,12 @@ void OnConnect(DbConnection conn, Identity identity, string authToken)
         .Subscribe("SELECT * FROM Message");
 }
 
-void OnConnectError(WebSocketError? error, string message)
+void OnConnectError(Exception e)
 {
 
 }
 
-void OnDisconnect(DbConnection conn, WebSocketCloseStatus? status, WebSocketError? error)
+void OnDisconnect(DbConnection conn, Exception? e)
 {
 
 }
diff --git a/src/WebSocket.cs b/src/WebSocket.cs
index 57f93af4..0ad50c06 100644
--- a/src/WebSocket.cs
+++ b/src/WebSocket.cs
@@ -9,8 +9,6 @@
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
-using Codice.Client.GameUI.Update;
-using UnityEngine;
 
 namespace SpacetimeDB
 {
@@ -82,7 +80,7 @@ public async Task Connect(string? auth, string host, string nameOrAddress, Addre
                 }
                 else
                 {
-                    if (OnConnectError != null) 
+                    if (OnConnectError != null)
                     {
                         dispatchQueue.Enqueue(() => OnConnectError(
                             new Exception($"WebSocket connection failed. Current state: {Ws.State}")));