diff --git a/src/components/StatusLine.tsx b/src/components/StatusLine.tsx
index 4cb853cc5..8544f759a 100644
--- a/src/components/StatusLine.tsx
+++ b/src/components/StatusLine.tsx
@@ -45,7 +45,7 @@ import { isVimModeEnabled } from './PromptInput/utils.js';
 import { computeHitRate, tokenSignature } from '../utils/cacheStats.js';
 import { onResponse as cacheOnResponse, getCacheStatsState, initCacheStatsState } from '../utils/cacheStatsState.js';
 import { BuiltinStatusLine } from './BuiltinStatusLine.js';
-import { formatTokens } from '../utils/format.js';
+import { formatTokens } from 'src/utils/format.js';
 
 // ---------------------------------------------------------------------------
 // CachePill — cache hit-rate + 1-hour TTL countdown pill
diff --git a/src/hooks/useGoalContinuation.ts b/src/hooks/useGoalContinuation.ts
index 5c02f6cbd..621e187bb 100644
--- a/src/hooks/useGoalContinuation.ts
+++ b/src/hooks/useGoalContinuation.ts
@@ -80,7 +80,13 @@ export function useGoalContinuation(opts: UseGoalContinuationOpts): void {
     }
 
     const goal = getGoal()
-    if (!goal) return
+    if (!goal) {
+      budgetLimitFiredRef.current = false
+      return
+    }
+    if (goal.status === 'active') {
+      budgetLimitFiredRef.current = false
+    }
 
     if (goal.status === 'budget_limited' && !budgetLimitFiredRef.current) {
       budgetLimitFiredRef.current = true