diff --git a/packages/yew-test-runner/src/procedural.rs b/packages/yew-test-runner/src/procedural.rs
index 972edb8d9a4..d4e74feb1bc 100644
--- a/packages/yew-test-runner/src/procedural.rs
+++ b/packages/yew-test-runner/src/procedural.rs
@@ -204,7 +204,10 @@ pub trait TestCase {
     /// # Example
     ///
     /// ```no_run
-    /// let test_runner = TestRunner::new();
+    /// use yew::html;
+    /// use yew_test_runner::procedural::{TestCase, TestRunner};
+    ///
+    /// let mut test_runner = TestRunner::new();
     /// test_runner
     ///     .render(html! {
     ///         <button>{"Click me!"}</button>