diff --git a/sao/src/view/tree.js b/sao/src/view/tree.js
index a45af17f78..8e6e64ccf8 100644
--- a/sao/src/view/tree.js
+++ b/sao/src/view/tree.js
@@ -1027,9 +1027,9 @@
                 var selector = `tr > td:nth-child(${i + offset + 1})`;
                 if ((thead_visible && column.header.is(':hidden')) ||
                     (column.header.css('display') == 'none')) {
-                    to_hide.add(this.body.find(selector));
+                    to_hide.add(this.tbody.find(selector));
                 } else {
-                    to_show.add(this.body.find(selector));
+                    to_show.add(this.tbody.find(selector));
                 }
             }