1 files changed, 6 insertions, 2 deletions
diff --git a/pure/src/widget/button.rs b/pure/src/widget/button.rs
index d0f9e53e..6dc1016c 100644
--- a/pure/src/widget/button.rs
+++ b/pure/src/widget/button.rs
@@ -85,8 +85,12 @@ where
         Box::new(State::new())
     }
 
-    fn children(&self) -> &[Element<Message, Renderer>] {
-        std::slice::from_ref(&self.content)
+    fn diff(&self, tree: &mut Tree) {
+        tree.diff_children(std::slice::from_ref(&self.content))
+    }
+
+    fn children_state(&self) -> Vec<Tree> {
+        vec![Tree::new(&self.content)]
     }
 
     fn width(&self) -> Length {