diff --git a/js/nodes_stats.js b/js/nodes_stats.js
index 3f84880..7ebd450 100644
--- a/js/nodes_stats.js
+++ b/js/nodes_stats.js
@@ -119,9 +119,7 @@ async function showStatsDialog() {
   dialog.querySelector("#ns-tab-nodes").addEventListener("click", () => switchTab("nodes"));
   dialog.querySelector("#ns-tab-models").addEventListener("click", () => switchTab("models"));
 
-  document
-    .getElementById("nodes-stats-close")
-    .addEventListener("click", () => overlay.remove());
+  dialog.querySelector("#nodes-stats-close").addEventListener("click", () => overlay.remove());
 
   // Toggle expandable rows
   dialog.querySelectorAll(".pkg-row").forEach((row) => {
diff --git a/tracker.py b/tracker.py
index 08e0908..5d78548 100644
--- a/tracker.py
+++ b/tracker.py
@@ -143,7 +143,8 @@ class UsageTracker:
                            VALUES (?, ?, 1, ?, ?)
                            ON CONFLICT(model_name) DO UPDATE SET
                                count = count + 1,
-                               last_seen = excluded.last_seen""",
+                               last_seen = excluded.last_seen,
+                               model_type = excluded.model_type""",
                         (model_name, model_type, now, now),
                     )
                 conn.commit()