1 files changed, 1 insertions, 1 deletions
diff --git a/web/app.js b/web/app.js
index 9742c25..adaa0a2 100644
--- a/web/app.js
+++ b/web/app.js
@@ -125,7 +125,7 @@ function createTaskCard(task) {
   }
 
   // Footer: action buttons based on state
-  const RESTART_STATES = new Set(['FAILED', 'CANCELLED']);
+  const RESTART_STATES = new Set(['FAILED', 'CANCELLED', 'BUDGET_EXCEEDED']);
   if (task.state === 'PENDING' || task.state === 'RUNNING' || task.state === 'READY' || task.state === 'BLOCKED' || task.state === 'TIMED_OUT' || RESTART_STATES.has(task.state)) {
     const footer = document.createElement('div');
     footer.className = 'task-card-footer';