diff --git a/Cargo.lock b/Cargo.lock
index 88102bb3fdc8..6ce7a67779e9 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -1952,6 +1952,7 @@ dependencies = [
  "datafusion-proto",
  "env_logger",
  "futures",
+ "itertools 0.14.0",
  "log",
  "mimalloc",
  "nix",
diff --git a/datafusion/common/src/dfschema.rs b/datafusion/common/src/dfschema.rs
index f0f0f3165f07..6be5446a0007 100644
--- a/datafusion/common/src/dfschema.rs
+++ b/datafusion/common/src/dfschema.rs
@@ -114,8 +114,7 @@ pub type DFSchemaRef = Arc<DFSchema>;
 /// To build a schema with metadata columns, use [DFSchema::new_with_metadata]:
 /// ```rust
 /// use datafusion_common::{DFSchema, Column, TableReference};
-/// use arrow_schema::{Field, Schema};
-/// use arrow::datatypes::DataType;
+/// use arrow::datatypes::{DataType, Field, Schema};
 /// use std::collections::HashMap;
 ///
 /// let schema = Schema::new(vec![
@@ -233,8 +232,7 @@ impl QualifiedSchema {
     /// ```rust
     /// use std::sync::Arc;
     /// use datafusion_common::{QualifiedSchema, TableReference};
-    /// use arrow_schema::{Field, Schema};
-    /// use arrow::datatypes::DataType;
+    /// use arrow::datatypes::{DataType, Field, Schema};
     /// let schema = Arc::new(Schema::new(vec![
     ///    Field::new("c1", DataType::Int32, false),
     /// ]));
@@ -388,8 +386,7 @@ impl QualifiedSchema {
     ///
     /// ```rust
     /// use datafusion_common::{DFSchema, QualifiedSchema, TableReference};
-    /// use arrow_schema::{Field, Schema};
-    /// use arrow::datatypes::DataType;
+    /// use arrow::datatypes::{DataType, Field, Schema};
     /// use std::sync::Arc;
     ///
     /// let schema1 = QualifiedSchema::new_with_table(
@@ -436,8 +433,7 @@ impl QualifiedSchema {
     ///
     /// ```
     /// use datafusion_common::{DFSchema, QualifiedSchema, TableReference};
-    /// use arrow_schema::{Field, Schema};
-    /// use arrow::datatypes::DataType;
+    /// use arrow::datatypes::{DataType, Field, Schema};
     /// use std::sync::Arc;
     ///
     /// let mut schema1 = QualifiedSchema::new_with_table(
diff --git a/datafusion/core/tests/sql/metadata_columns.rs b/datafusion/core/tests/sql/metadata_columns.rs
index c5f7e16c1903..c39a2e5f81e8 100644
--- a/datafusion/core/tests/sql/metadata_columns.rs
+++ b/datafusion/core/tests/sql/metadata_columns.rs
@@ -19,7 +19,7 @@ use std::any::Any;
 use std::fmt::{self, Debug, Formatter};
 use std::sync::{Arc, Mutex};
 
-use arrow_array::{ArrayRef, StringArray, UInt64Array};
+use arrow::array::{ArrayRef, StringArray, UInt64Array};
 use async_trait::async_trait;
 use datafusion::arrow::array::{UInt64Builder, UInt8Builder};
 use datafusion::arrow::datatypes::{DataType, Field, Schema, SchemaRef};