diff --git a/src/lib/ChatRequestPetals.svelte b/src/lib/ChatRequestPetals.svelte
index 295a37b..8df2175 100644
--- a/src/lib/ChatRequestPetals.svelte
+++ b/src/lib/ChatRequestPetals.svelte
@@ -59,6 +59,7 @@ export const runPetalsCompletionRequest = async (
         }
         chatRequest.updating = false
         chatRequest.updatingMessage = ''
+        ws.close()
       })
       ws.onopen = () => {
         ws.send(JSON.stringify({