         expect(tableElement.getLastVisibleRow()).toEqual(23)
 
     it 'renders new rows', ->
-      rows = tableElement.find('.table-edit-row')
+      rows = tableShadowRoot.querySelectorAll('.table-edit-row')
       expect(rows.length).toEqual(28)
       expect(rows.first().data('row-id')).toEqual(6)
       expect(rows.last().data('row-id')).toEqual(33)


      nextAnimationFrame()

      expect(tableElement.find('.selected').length).toEqual(6)

    it 'marks the row number with a selected class', ->
      tableElement.showGutter()

      nextAnimationFrame()

    it 'activates the cell under the mouse when pressed', ->
      cell = tableShadowRoot.querySelectorAll('.table-edit-row:nth-child(4) .table-edit-cell:last-child')
      mousedown(cell)

      expect(tableElement.getActiveCell().getValue()).toEqual('no')
