From e21ee6d1fd7be6c55beaa5d09589cbf2402fb186 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Thu, 11 Sep 2025 15:55:04 +0100
Subject: [PATCH] Await async websocket closure

---
 trustgraph-flow/trustgraph/gateway/dispatch/mux.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/trustgraph-flow/trustgraph/gateway/dispatch/mux.py b/trustgraph-flow/trustgraph/gateway/dispatch/mux.py
index afce6b75..ddaa8ddf 100644
--- a/trustgraph-flow/trustgraph/gateway/dispatch/mux.py
+++ b/trustgraph-flow/trustgraph/gateway/dispatch/mux.py
@@ -147,7 +147,7 @@ class Mux:
                 self.running.stop()
 
                 if self.ws:
-                    self.ws.close()
+                    await self.ws.close()
                     self.ws = None
 
                 break
@@ -165,6 +165,6 @@ class Mux:
         self.running.stop()
 
         if self.ws:
-            self.ws.close()
+            await self.ws.close()
             self.ws = None