diff --git a/rust/src/lib.rs b/rust/src/lib.rs
index 27ef2841b7..9aece98ff2 100644
--- a/rust/src/lib.rs
+++ b/rust/src/lib.rs
@@ -52,10 +52,10 @@
 //! Querying from local filesystem:
 //! ```ignore
 //! use std::sync::Arc;
-//! use datafusion::execution::context::ExecutionContext;
+//! use datafusion::prelude::SessionContext;
 //!
 //! async {
-//!   let mut ctx = ExecutionContext::new();
+//!   let mut ctx = SessionContext::new();
 //!   let table = deltalake::open_table("./tests/data/simple_table")
 //!       .await
 //!       .unwrap();