diff --git a/tokio/src/runtime/dump.rs b/tokio/src/runtime/dump.rs
index 5f07ef70862..6c5e380c604 100644
--- a/tokio/src/runtime/dump.rs
+++ b/tokio/src/runtime/dump.rs
@@ -227,6 +227,12 @@ impl Trace {
     ///
     /// Example usage:
     /// ```
+    /// # use std::future::Future;
+    /// # use std::task::Poll;
+    ///
+    /// # use tokio::runtime::dump::{Root, Trace};
+    ///
+    /// # async fn test_fn() {
     /// // some future
     /// let mut test_future = std::pin::pin!(async move { tokio::task::yield_now().await; 0 });
     ///
@@ -242,6 +248,7 @@ impl Trace {
     /// };
     ///
     /// println!("{trace}");
+    /// # }
     /// ```
     ///
     /// ### Nested calls