diff --git a/src/layout/widget.rs b/src/layout/widget.rs
index afe95bc..d00d924 100644
--- a/src/layout/widget.rs
+++ b/src/layout/widget.rs
@@ -33,6 +33,12 @@ pub trait WidgetLike<Ctx, Tag> {
     {
         self.add(ui).into_static()
     }
+    fn set_root(self, ui: &mut Ui<Ctx>)
+    where
+        Self: Sized,
+    {
+        ui.set_root(self);
+    }
 }
 
 /// A function that returns a widget given a UI.
diff --git a/src/testing/mod.rs b/src/testing/mod.rs
index c530b2b..76caa69 100644
--- a/src/testing/mod.rs
+++ b/src/testing/mod.rs
@@ -183,13 +183,13 @@ impl Client {
 
         let info = text("").add(&mut ui);
         let info_sect = info.clone().pad(10).align(Align::BotLeft);
-        ui.set_root(
-            (
-                (tabs, main).span(Dir::DOWN, [fixed(40), ratio(1)]),
-                info_sect,
-            )
-                .stack(),
-        );
+
+        (
+            (tabs, main).span(Dir::DOWN, [fixed(40), ratio(1)]),
+            info_sect,
+        )
+            .stack()
+            .set_root(&mut ui);
 
         Self {
             renderer,