diff --git a/package.json b/package.json
index bad9f97..80c4236 100644
--- a/package.json
+++ b/package.json
@@ -1,7 +1,7 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
   "name": "@troed/oc-ls-stats",
-  "version": "0.0.41",
+  "version": "0.0.42",
   "type": "module",
   "exports": {
     "./tui": {
diff --git a/tui.tsx b/tui.tsx
index b074c34..8b23310 100644
--- a/tui.tsx
+++ b/tui.tsx
@@ -328,6 +328,7 @@ const tui: TuiPlugin = async (api) => {
 
     let anyPrefilling = false
     let anyGenerating = false
+    let genEnded = false
 
     for (const baseUrl of llamaServerUrls) {
       const slots = await fetchSlots(baseUrl, model)
@@ -386,6 +387,9 @@ const tui: TuiPlugin = async (api) => {
             const delta = nd - tracker.generateCapturedDecoded
             if (dt > 0 && delta > 0) {
               tracker.lastGeneratedTps = delta / dt
+            } else if (delta <= 0) {
+              tracker.lastGeneratedTps = 0
+              genEnded = true
             }
           }
         } else {
@@ -399,6 +403,8 @@ const tui: TuiPlugin = async (api) => {
           tracker.generateSlotId = null
           tracker.generateCapturedDecoded = null
           tracker.generateStartAt = 0
+          tracker.lastGeneratedTps = 0
+          genEnded = true
         }
       }
 
@@ -406,7 +412,7 @@ const tui: TuiPlugin = async (api) => {
       tracker.isGenerating = anyGenerating
     }
 
-    if (tracker.lastPrefillRate > 0 || tracker.lastGeneratedTps > 0) {
+    if (tracker.lastPrefillRate > 0 || tracker.lastGeneratedTps > 0 || genEnded) {
       bump()
     }
   }