1 files changed, 6 insertions, 2 deletions
diff --git a/pure/src/widget/container.rs b/pure/src/widget/container.rs
index 94a6b07b..85ea8039 100644
--- a/pure/src/widget/container.rs
+++ b/pure/src/widget/container.rs
@@ -132,8 +132,12 @@ where
         Box::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 {