diff --git a/x-wing/src/lib.rs b/x-wing/src/lib.rs
index 588b7f0..7284315 100644
--- a/x-wing/src/lib.rs
+++ b/x-wing/src/lib.rs
@@ -14,16 +14,12 @@
 //! decapsulation key. X-Wing-KEM is a general-purpose hybrid post-quantum KEM, combining x25519 and ML-KEM-768.
 //!
 //! ```
-//! # use rand;
-//! # use kem::{Decapsulate, Encapsulate};
-//! let mut rng = rand::thread_rng();
-//!
-//! let (sk, pk) = x_wing::generate_key_pair(&mut rng);
-//!
-//! let (ct, ss_sender) = pk.encapsulate(&mut rng).unwrap();
+//! use kem::{Decapsulate, Encapsulate};
 //!
+//! let mut rng = &mut rand::rngs::OsRng;
+//! let (sk, pk) = x_wing::generate_key_pair(rng);
+//! let (ct, ss_sender) = pk.encapsulate(rng).unwrap();
 //! let ss_receiver = sk.decapsulate(&ct).unwrap();
-//!
 //! assert_eq!(ss_sender, ss_receiver);
 //! ```