diff --git a/crates/agent/src/active_thread.rs b/crates/agent/src/active_thread.rs
index 01195552a6..140294cbbc 100644
--- a/crates/agent/src/active_thread.rs
+++ b/crates/agent/src/active_thread.rs
@@ -2082,7 +2082,7 @@ impl ActiveThread {
                 v_flex()
                     .w_full()
                     .map(|parent| {
-                        if let Some(checkpoint) = checkpoint.filter(|_| is_generating) {
+                        if let Some(checkpoint) = checkpoint.filter(|_| !is_generating) {
                             let mut is_pending = false;
                             let mut error = None;
                             if let Some(last_restore_checkpoint) =