     it "restores the focus state of the tree view", ->
       rootView.attachToDom()
       treeView.focus()
-      expect(treeView.find(".tree-view")).toMatchSelector ':focus'
+      expect(treeView.treeViewList).toMatchSelector ':focus'
       atom.deactivatePackage("tree-view")
       atom.activatePackage("tree-view")
       treeView = rootView.find(".tree-view").view()
-      expect(treeView.find(".tree-view")).toMatchSelector ':focus'
+      expect(treeView.treeViewList).toMatchSelector ':focus'
 
     it "restores the scroll top when toggled", ->
       rootView.height(5)

        treeView.height(100)
        treeView.attachToDom()
        $(element).view().expand() for element in treeView.find('.directory')
        expect(treeView.find(".tree-view").prop('scrollHeight')).toBeGreaterThan treeView.find(".tree-view").outerHeight()

        treeView.moveDown()
        expect(treeView.scrollTop()).toBe 0

        treeView.height(100)
        treeView.attachToDom()
        $(element).view().expand() for element in treeView.find('.directory')
        expect(treeView.treeViewList.prop('scrollHeight')).toBeGreaterThan treeView.treeViewList.outerHeight()

        expect(treeView.scrollTop()).toBe 0
        treeView.trigger 'core:move-to-bottom'
