diff --git a/Cargo.toml b/Cargo.toml
index ee3ca8a..26a20b1 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -7,7 +7,7 @@ repository = "https://github.com/sundy-li/arrow_cli"
 edition = "2024"
 license = "Apache-2.0"
 name = "arrow_cli"
-version = "0.2.1"
+version = "0.2.2"
 
 
 [dependencies]
diff --git a/src/session.rs b/src/session.rs
index e39a2d5..0adda4b 100644
--- a/src/session.rs
+++ b/src/session.rs
@@ -2,7 +2,9 @@ use arrow::csv::WriterBuilder;
 use arrow::error::ArrowError;
 use arrow::record_batch::RecordBatch;
 use arrow_cast::pretty::pretty_format_batches;
-use arrow_flight::sql::client::FlightSqlServiceClient;
+use arrow_flight::{
+    flight_service_client::FlightServiceClient, sql::client::FlightSqlServiceClient,
+};
 use futures::TryStreamExt;
 use rustyline::Editor;
 use rustyline::error::ReadlineError;
@@ -36,7 +38,10 @@ impl Session {
             println!("Connecting to {} as user {}.", endpoint.uri(), user);
             println!();
         }
-        let mut client = FlightSqlServiceClient::new(channel);
+
+        let mut client = FlightSqlServiceClient::new_from_inner(
+            FlightServiceClient::new(channel).max_decoding_message_size(usize::MAX),
+        );
         let _token = client.handshake(user, password).await?;
 
         let prompt = format!("{} :) ", endpoint.uri().host().unwrap());