diff --git a/test/tests/libraryTreeViewTest.js b/test/tests/libraryTreeViewTest.js index 00f4e6201..91d6b4c60 100644 --- a/test/tests/libraryTreeViewTest.js +++ b/test/tests/libraryTreeViewTest.js @@ -31,15 +31,15 @@ describe("Zotero.LibraryTreeView", function() { it("should remove the last row", function* () { var collection = yield createDataObject('collection'); yield waitForItemsLoad(win); - var item1 = yield createDataObject('item', { collections: [collection.id] }); - var item2 = yield createDataObject('item', { collections: [collection.id] }); + yield createDataObject('item', { collections: [collection.id] }); + yield createDataObject('item', { collections: [collection.id] }); var view = zp.itemsView; - assert.equal(view.getRowIndexByID(item2.id), 1); + var treeViewID = view.getRow(1).id; zp.itemsView._removeRow(1); assert.equal(view.rowCount, 1); - assert.isFalse(view.getRowIndexByID(item2.id)); + assert.isFalse(view.getRowIndexByID(treeViewID)); }); }); })