From b534664534b316d35266381e210bb66be1f17a99 Mon Sep 17 00:00:00 2001
From: CaliBrain <calibrain@l4n.xyz>
Date: Fri, 10 Jan 2025 12:57:42 -0500
Subject: [PATCH] Fix status not getting removed properly (#51)

---
 models.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/models.py b/models.py
index e703914d..d086ed8f 100644
--- a/models.py
+++ b/models.py
@@ -90,8 +90,7 @@ class BookQueue:
                 # Check for stale status entries
                 last_update = self._status_timestamps.get(book_id)
                 if last_update and (current_time - last_update) > self._status_timeout:
-                    # Don't remove DONE status
-                    if status == QueueStatus.DONE:
+                    if status == QueueStatus.DONE or status == QueueStatus.ERROR or status == QueueStatus.AVAILABLE:
                         to_remove.append(book_id)
             
             # Remove stale entries