Fix deferred layout on resize after drawing

1 files changed, 13 insertions, 10 deletions

diff --git a/widget/src/lazy/responsive.rs b/widget/src/lazy/responsive.rs
index a8abbce8..f9bd0334 100644
--- a/widget/src/lazy/responsive.rs
+++ b/widget/src/lazy/responsive.rs
@@ -82,18 +82,21 @@ where
         new_size: Size,
         view: &dyn Fn(Size) -> Element<'a, Message, Theme, Renderer>,
     ) {
-        let is_tree_empty =
-            tree.tag == tree::Tag::stateless() && tree.children.is_empty();
+        if self.size != new_size {
+            self.element = view(new_size);
+            self.size = new_size;
+            self.layout = None;
 
-        if !is_tree_empty && self.size == new_size {
-            return;
-        }
-
-        self.element = view(new_size);
-        self.size = new_size;
-        self.layout = None;
+            tree.diff(&self.element);
+        } else {
+            let is_tree_empty =
+                tree.tag == tree::Tag::stateless() && tree.children.is_empty();
 
-        tree.diff(&self.element);
+            if is_tree_empty {
+                self.layout = None;
+                tree.diff(&self.element);
+            }
+        }
     }
 
     fn resolve<R, T>(