diff --git a/crates/zeta_cli/src/main.rs b/crates/zeta_cli/src/main.rs
index b2822c2e1efb31dc4149f9e8814ab6664d334796..b8e7c9f5e1df79aaf9ad0485838d7db65dfabce0 100644
--- a/crates/zeta_cli/src/main.rs
+++ b/crates/zeta_cli/src/main.rs
@@ -450,6 +450,7 @@ async fn zeta2_predict(
                         &mut excerpts_text,
                     );
                 }
+                break;
             }
             _ => {}
         }
@@ -771,6 +772,7 @@ fn main() {
                     Zeta2Command::Predict { example_path } => {
                         let example = NamedExample::load(example_path).unwrap();
                         zeta2_predict(example, &app_state, cx).await.unwrap();
+                        let _ = cx.update(|cx| cx.quit());
                         return;
                     }
                     Zeta2Command::Syntax {