diff --git a/StabilityMatrix.Core/Inference/ComfyClient.cs b/StabilityMatrix.Core/Inference/ComfyClient.cs
index 5f217b72..f4b21b6a 100644
--- a/StabilityMatrix.Core/Inference/ComfyClient.cs
+++ b/StabilityMatrix.Core/Inference/ComfyClient.cs
@@ -80,9 +80,11 @@ public class ComfyClient : InferenceClientBase
             Path = "/ws",
             Query = $"clientId={ClientId}"
         }.Uri;
+
         webSocketClient = new WebsocketClient(wsUri)
         {
-            ReconnectTimeout = TimeSpan.FromSeconds(30),
+            Name = "ComfyClient",
+            ReconnectTimeout = TimeSpan.FromSeconds(30)
         };
 
         webSocketClient.DisconnectionHappened.Subscribe(