diff --git a/stomp.js b/stomp.js
index 9a927c6..12319ed 100644
--- a/stomp.js
+++ b/stomp.js
@@ -40,9 +40,9 @@ function doc_keyUp(e) {
 
 
     var startpos;
-    var nodes = document.getElementsByClassName("table-row leaf first")[0].parentNode.children;
+    var nodes = document.getElementsByClassName("table-row first")[0].parentNode.children;
     for (let i = 0; i < nodes.length; i++) {
-      if (nodes[i].className.match(/table-row leaf (?:selected select-single )?first/g)) {
+      if (nodes[i].className.match(/table-row (?:selected select-single )?first/g)) {
         startpos = i;
         break;
       }