diff --git a/src/tree.rs b/src/tree.rs
index dde860a..1fdfa20 100644
--- a/src/tree.rs
+++ b/src/tree.rs
@@ -181,7 +181,7 @@ impl Tree {
 
             let mut index = 0;
             while index < unordered.len() {
-                let layout = match &data.nodes[unordered[index]].layout {
+                let layout = match data.nodes.get(unordered[index]).and_then(|n| n.layout) {
                     Some(layout) if layout.size.width > 0 && layout.size.height > 0 => layout,
                     _ => {
                         unordered.remove(index);