diff --git a/web/comfy_shared.js b/web/comfy_shared.js
index 433ed95..c96fbdd 100644
--- a/web/comfy_shared.js
+++ b/web/comfy_shared.js
@@ -30,7 +30,7 @@ export const cleanupNode = (node) => {
     return
   }
 
-  for (const w of this.widgets) {
+  for (const w of node.widgets) {
     if (w.canvas) {
       w.canvas.remove()
     }