1 files changed, 4 insertions, 2 deletions
diff --git a/src/server-component.mjs b/src/server-component.mjs
index 373f4cd..ab08801 100644
--- a/src/server-component.mjs
+++ b/src/server-component.mjs
@@ -136,13 +136,15 @@ export class LustreServerComponent extends HTMLElement {
   }
 
   morph(vdom) {
-    this.#root = morph(this.#root, vdom, (msg) => {
+    this.#root = morph(this.#root, vdom, (handler) => (event) => {
+      const msg = handler(event);
       this.#socket?.send(JSON.stringify([Constants.event, msg.tag, msg.data]));
     });
   }
 
   diff([diff]) {
-    this.#root = patch(this.#root, diff, (msg) => {
+    this.#root = patch(this.#root, diff, (handler) => (event) => {
+      const msg = handler(event);
       this.#socket?.send(JSON.stringify([Constants.event, msg.tag, msg.data]));
     });
   }