diff --git a/src/expr/udf/src/error.rs b/src/expr/udf/src/error.rs
index 17306f172970d..58e22c1b79305 100644
--- a/src/expr/udf/src/error.rs
+++ b/src/expr/udf/src/error.rs
@@ -57,7 +57,11 @@ impl Error {
     }
 
     pub fn is_tonic_error(&self) -> bool {
-        matches!(self.inner(), ErrorInner::Tonic(_))
+        match self.inner() {
+            ErrorInner::Tonic(_) => true,
+            ErrorInner::Flight(flight_error) => matches!(flight_error, FlightError::Tonic(_)),
+            _ => false,
+        }
     }
 }