diff --git a/src/tqdm_publisher/_demo/_client.html b/src/tqdm_publisher/_demo/_client.html
index 0e01ef3..65d60c4 100644
--- a/src/tqdm_publisher/_demo/_client.html
+++ b/src/tqdm_publisher/_demo/_client.html
@@ -120,9 +120,8 @@ <h1>tqdm_progress</h1>
     onopen: () => console.log('Connected'),
     onclose: () => console.log('Disconnected'),
     onmessage: (data) => {
-        const { id, payload } = JSON.parse(event.data);
-        console.log(id, payload, bars[id]);
-        bars[id].style.width = 100 * (payload.n / payload.total) + '%';
+        const { progress_bar_id, format_dict } = JSON.parse(event.data);
+        bars[progress_bar_id].style.width = 100 * (format_dict.n / format_dict.total) + '%';
     }
 });