diff --git a/js/nodes_stats.js b/js/nodes_stats.js
index c12de68..43a6b67 100644
--- a/js/nodes_stats.js
+++ b/js/nodes_stats.js
@@ -14,33 +14,17 @@ app.registerExtension({
   name: "comfyui.nodes_stats",
 
   async setup() {
-    try {
-      const { ComfyButton } = await import(
-        "../../scripts/ui/components/button.js"
-      );
-
-      const btn = new ComfyButton({
-        icon: "bar-chart-2",
-        content: "Node Stats",
-        tooltip: "Show node and package usage statistics",
-        action: () => showStatsDialog(),
-        classList: "comfyui-button comfyui-menu-mobile-collapse",
-      });
-
-      app.menu?.settingsGroup.element.before(btn.element);
-    } catch (e) {
-      console.log(
-        "[nodes-stats] New menu API unavailable, falling back to legacy menu",
-        e
-      );
-
-      const btn = document.createElement("button");
-      btn.innerHTML = STATS_ICON;
-      btn.title = "Node Stats";
-      btn.onclick = () => showStatsDialog();
-      btn.style.cssText =
-        "display:flex;align-items:center;justify-content:center;padding:6px;background:none;border:none;cursor:pointer;color:var(--input-text,#ddd);";
+    const btn = document.createElement("button");
+    btn.innerHTML = STATS_ICON;
+    btn.title = "Node Stats";
+    btn.className = "comfyui-button comfyui-menu-mobile-collapse";
+    btn.onclick = () => showStatsDialog();
+    btn.style.cssText =
+      "display:flex;align-items:center;justify-content:center;padding:6px;cursor:pointer;";
 
+    if (app.menu?.settingsGroup?.element) {
+      app.menu.settingsGroup.element.before(btn);
+    } else {
       const menu = document.querySelector(".comfy-menu");
       if (menu) {
         menu.append(btn);