diff --git a/desktop/treeview.c b/desktop/treeview.c
index 62e196ee2..1de4bf968 100644
--- a/desktop/treeview.c
+++ b/desktop/treeview.c
@@ -2321,6 +2321,7 @@ nserror treeview_contract(treeview *tree, bool all)
 	r.x1 = REDRAW_MAX;
 	r.y1 = tree->root->height;
 
+	data.tree = tree;
 	data.only_entries = !all;
 
 	for (n = tree->root->children; n != NULL; n = n->next_sib) {