diff --git a/mail/base/content/mailContextMenus.js b/mail/base/content/mailContextMenus.js
index bb804f4e9462..e21f428c2045 100644
--- a/mail/base/content/mailContextMenus.js
+++ b/mail/base/content/mailContextMenus.js
@@ -43,8 +43,7 @@ function RestoreSelectionWithoutContentLoad(tree)
       return;
     }
 
-    var treeBoxObj = tree.treeBoxObject;
-    var treeSelection = treeBoxObj.selection;
+    var treeSelection = tree.view.selection;
 
     // make sure that currentIndex is valid so that we don't try to restore
     // a selection of an invalid row.
@@ -263,7 +262,7 @@ function fillFolderPaneContextMenu()
   var folderTree = GetFolderTree();
   var startIndex = {};
   var endIndex = {};
-  folderTree.treeBoxObject.selection.getRangeAt(0, startIndex, endIndex);
+  folderTree.view.selection.getRangeAt(0, startIndex, endIndex);
   if (startIndex.value < 0)
     return false;
   var numSelected = endIndex.value - startIndex.value + 1;