diff --git a/src/case/tree/treeexpander/tree.expander.js b/src/case/tree/treeexpander/tree.expander.js
index 50d2bef61..88f5dbf50 100644
--- a/src/case/tree/treeexpander/tree.expander.js
+++ b/src/case/tree/treeexpander/tree.expander.js
@@ -67,6 +67,10 @@
         populate: function (items) {
             this.expander.populate(items);
         },
+
+        getAllLeaves: function () {
+            return this.expander && this.expander.getAllLeaves();
+        }
     });
 
     BI.shortcut("bi.tree_expander", Widget);
diff --git a/src/case/tree/treeexpander/tree.expander.popup.js b/src/case/tree/treeexpander/tree.expander.popup.js
index a101b728c..6deae0982 100644
--- a/src/case/tree/treeexpander/tree.expander.popup.js
+++ b/src/case/tree/treeexpander/tree.expander.popup.js
@@ -43,6 +43,10 @@
         populate: function (items) {
             this.popupview.populate(items);
         },
+
+        getAllLeaves: function () {
+            return this.popupView && this.popupView.getAllLeaves();
+        }
     });
 
     BI.shortcut("bi.tree_expander.popup", Widget);