diff --git a/StabilityMatrix.Core/Inference/ComfyClient.cs b/StabilityMatrix.Core/Inference/ComfyClient.cs
index 28dc53a2..be77a61c 100644
--- a/StabilityMatrix.Core/Inference/ComfyClient.cs
+++ b/StabilityMatrix.Core/Inference/ComfyClient.cs
@@ -261,7 +261,7 @@ public class ComfyClient : InferenceClientBase
     public override async Task CloseAsync(CancellationToken cancellationToken = default)
     {
         await webSocketClient
-            .StopOrFail(WebSocketCloseStatus.NormalClosure, string.Empty)
+            .Stop(WebSocketCloseStatus.NormalClosure, string.Empty)
             .ConfigureAwait(false);
     }