From 20b044865c9b5aa41abbe97a94e75a313024d414 Mon Sep 17 00:00:00 2001
From: Shadow Cat <shadowcatzero@protonmail.com>
Date: Sat, 20 Sep 2025 13:42:47 -0400
Subject: [PATCH] we love post fix

---
 src/layout/widget.rs |  6 ++++++
 src/testing/mod.rs   | 14 +++++++-------
 2 files changed, 13 insertions(+), 7 deletions(-)

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,