diff --git a/calendar/resources/content/unifinder.js b/calendar/resources/content/unifinder.js
index 0e230fcdae32..eb0ea92962f7 100644
--- a/calendar/resources/content/unifinder.js
+++ b/calendar/resources/content/unifinder.js
@@ -475,7 +475,8 @@ var treeView =
    
       if (sortActive != "true")
       {
-         treeCols = document.getElementsByTagName("treecol");
+         var unifinder = document.getElementById("unifinder-search-results-listbox");
+         treeCols = unifinder.getElementsByTagName("treecol");
          for (var i = 0; i < treeCols.length; i++)
          {
             treeCols[i].removeAttribute("sortActive");
diff --git a/calendar/resources/content/unifinderToDo.js b/calendar/resources/content/unifinderToDo.js
index 86a3dc255c46..701816793304 100644
--- a/calendar/resources/content/unifinderToDo.js
+++ b/calendar/resources/content/unifinderToDo.js
@@ -359,7 +359,8 @@ var toDoTreeView =
    
       if (sortActive != "true")
       {
-         treeCols = document.getElementsByTagName("treecol");
+         var taskList = document.getElementById("unifinder-todo-tree");
+         treeCols = taskList.getElementsByTagName("treecol");
          for (var i = 0; i < treeCols.length; i++)
          {
             treeCols[i].removeAttribute("sortActive");