diff --git a/static/client.mjs b/static/client.mjs
index 1235142..863ee24 100644
--- a/static/client.mjs
+++ b/static/client.mjs
@@ -25,7 +25,7 @@ try {
 
 const ws = new WebSocket(`ws://localhost:${window.VIV_PORT}`);
 
-ws.addEventListener('message', (event) => {
+ws.addEventListener('message', async (event) => {
     const fields = event.data.toString().split(': ');
     if (fields.length < 2) return;
     const [key, ...values] = fields;
@@ -34,9 +34,7 @@ ws.addEventListener('message', (event) => {
     switch (key) {
         case 'UPDATE':
             document.getElementById('body-content').innerHTML = value;
-            (async () => {
-                await mermaid.run({ querySelector: '.mermaid' });
-            })();
+            await mermaid.run({ querySelector: '.mermaid' });
             break;
 
         case 'SCROLL':