diff --git a/src/widget/context_menu.rs b/src/widget/context_menu.rs
index dfec06d..261779f 100644
--- a/src/widget/context_menu.rs
+++ b/src/widget/context_menu.rs
@@ -91,7 +91,7 @@ impl<'a, Message: Clone> Widget<Message, crate::Theme, crate::Renderer>
     }
 
     fn diff(&mut self, tree: &mut Tree) {
-        self.content.as_widget_mut().diff(&mut tree.children[0]);
+        tree.children[0].diff(self.content.as_widget_mut());
 
         // if let Some(ref mut context_menus) = self.context_menu {
         //     for (menu, tree) in context_menus
diff --git a/src/widget/dnd_destination.rs b/src/widget/dnd_destination.rs
index 57a5276..0aca0d7 100644
--- a/src/widget/dnd_destination.rs
+++ b/src/widget/dnd_destination.rs
@@ -253,7 +253,7 @@ impl<'a, Message: 'static> Widget<Message, crate::Theme, crate::Renderer>
     }
 
     fn diff(&mut self, tree: &mut Tree) {
-        self.container.as_widget_mut().diff(&mut tree.children[0]);
+        tree.children[0].diff(self.container.as_widget_mut());
     }
 
     fn state(&self) -> iced_core::widget::tree::State {
diff --git a/src/widget/dnd_source.rs b/src/widget/dnd_source.rs
index e1699ed..5a37820 100644
--- a/src/widget/dnd_source.rs
+++ b/src/widget/dnd_source.rs
@@ -132,7 +132,7 @@ impl<
     }
 
     fn diff(&mut self, tree: &mut Tree) {
-        self.container.as_widget_mut().diff(&mut tree.children[0]);
+        tree.children[0].diff(self.container.as_widget_mut());
     }
 
     fn state(&self) -> iced_core::widget::tree::State {