diff --git a/crates/ui/src/main.rs b/crates/ui/src/main.rs
index 4786c58..e8e719d 100644
--- a/crates/ui/src/main.rs
+++ b/crates/ui/src/main.rs
@@ -292,7 +292,7 @@ fn AskPanel(datasets: Vec<Dataset>) -> Element {
 
                     // Step 3b: If schema error, retry with the error as feedback
                     if let Err(ref err) = query_result {
-                        if err.contains("Schema error") || err.contains("No field named") {
+                        if err.contains("error") {
                             step.set("fixing SQL...".into());
                             let retry_prompt = format!(
                                 "The SQL you wrote had an error:\n{err}\n\n\
@@ -368,7 +368,7 @@ fn AskPanel(datasets: Vec<Dataset>) -> Element {
                                         step.set("running query...".into());
                                         let query_result = run_sql(&sql).await;
                                         if let Err(ref err) = query_result {
-                                            if err.contains("Schema error") || err.contains("No field named") {
+                                            if err.contains("error") {
                                                 step.set("fixing SQL...".into());
                                                 let retry_prompt = format!(
                                                     "The SQL you wrote had an error:\n{err}\n\n{schema_ctx}\n\nOriginal question: {q}\n\nWrite a CORRECTED SQL query using ONLY the columns listed. Output ONLY SQL."