diff --git a/web/scripts/api.js b/web/scripts/api.js
index 87563a435..670d32842 100644
--- a/web/scripts/api.js
+++ b/web/scripts/api.js
@@ -47,7 +47,7 @@ class ComfyApi extends EventTarget {
 						this.dispatchEvent(new CustomEvent("executed", { detail: msg.data }));
 						break;
 				}
-				lastMessage = i;
+				lastMessage = msg.poll_id;
 			});
 		}, 2000)
 	}