 
     describe 'by updating the content of a row', ->
       it 'update the rows', ->
-        rows = tableElement.find('.table-edit-row')
+        rows = tableShadowRoot.querySelectorAll('.table-edit-row')
         expect(rows.eq(6).find('.table-edit-cell').first().text()).toEqual('row6')
 
         table.getRow(6).key = 'foo'
 
         nextAnimationFrame()
 
-        rows = tableElement.find('.table-edit-row')
+        rows = tableShadowRoot.querySelectorAll('.table-edit-row')
         expect(rows.eq(6).find('.table-edit-cell').first().text()).toEqual('foo')
 
   describe 'setting a custom height for a row', ->

          atom.commands.dispatch(editor.element, 'core:confirm')

        it 'closes the editor', ->
          expect(tableElement.find('atom-text-editor:visible').length).toEqual(0)

        it 'gives the focus back to the table view', ->
          expect(tableElement.hiddenInput.is(':focus')).toBeTruthy()

          nextAnimationFrame()

          expect(tableElement.direction).toEqual(1)
          expect(tableShadowRoot.querySelectorAll('.table-edit-row:first-child .table-edit-cell:first-child').text()).toEqual('row0')

      describe '::resetSort', ->
        beforeEach ->
