1 files changed, 10 insertions, 1 deletions
diff --git a/client/src/window.rs b/client/src/window.rs
index 5f2e7f8..2219bbc 100644
--- a/client/src/window.rs
+++ b/client/src/window.rs
@@ -43,7 +43,7 @@ impl ApplicationHandler for WindowState {
                 WindowEvent::Resized(size) => {
                     sta.renderer.resize(size.width, size.height);
                 }
-                WindowEvent::RedrawRequested => sta.renderer.draw().unwrap(),
+                WindowEvent::RedrawRequested => sta.renderer.draw(&sta.tree).unwrap(),
                 WindowEvent::CloseRequested => {
                     event_loop.exit();
                 }
@@ -59,3 +59,12 @@ impl ApplicationHandler for WindowState {
         }
     }
 }
+
+impl Drop for WindowState {
+    fn drop(&mut self) {
+        if let Some((win, sta)) = self.window.take() {
+            drop(sta);
+            drop(win)
+        }
+    }
+}