diff --git a/content/treestyletab/editBookmarkOverlay.js b/content/treestyletab/editBookmarkOverlay.js
index d623d125..47ac885e 100644
--- a/content/treestyletab/editBookmarkOverlay.js
+++ b/content/treestyletab/editBookmarkOverlay.js
@@ -83,7 +83,7 @@ var TreeStyleTabBookmarksProperty = {
 		var items = this._getItemsInFolder(PlacesUtils.bookmarks.getFolderIdForItem(aCurrentItem));
 		var treeStructure = TreeStyleTabService.getTreeStructureFromBookmarkItems(items);
 
-		var selected = treeStructure[current];
+		var selected = treeStructure[items.indexOf(aCurrentItem)];
 		if (selected > -1) selected = items[selected];
 
 		var fragment = document.createDocumentFragment();