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