diff --git a/web/js/main.js b/web/js/main.js
index 310b798..5f0998c 100644
--- a/web/js/main.js
+++ b/web/js/main.js
@@ -352,6 +352,7 @@ function handleWebSocketMessage(event) {
     const sum = signalData.reduce((acc, strNum) => acc + parseFloat(strNum), 0);
     data.push(sum / signalData.length);
 }
+socket.onmessage = handleWebSocketMessage;
 
 const signalBuffer = [];