     expect(activeCell.getValue()).toEqual('row0')
 
   it 'renders the active cell using a class', ->
-    expect(tableElement.find('.table-edit-header-cell.active-column').length).toEqual(1)
-    expect(tableElement.find('.table-edit-row.active-row').length).toEqual(1)
-    expect(tableElement.find('.table-edit-cell.active').length).toEqual(1)
-    expect(tableElement.find('.table-edit-cell.active-column').length)
+    expect(tableShadowRoot.querySelectorAll('.table-edit-header-cell.active-column').length).toEqual(1)
+    expect(tableShadowRoot.querySelectorAll('.table-edit-row.active-row').length).toEqual(1)
+    expect(tableShadowRoot.querySelectorAll('.table-edit-cell.active').length).toEqual(1)
+    expect(tableShadowRoot.querySelectorAll('.table-edit-cell.active-column').length)
     .toBeGreaterThan(1)
 
   describe 'when the absoluteColumnsWidths setting is enabled', ->


    describe 'clicking on another cell', ->
      beforeEach ->
        cell = tableElement.find('.table-edit-row:nth-child(4) .table-edit-cell:last-child')
        mousedown(cell)

      it 'closes the editor', ->

      [column] = []

      beforeEach ->
        column = tableShadowRoot.querySelectorAll('.table-edit-header-cell:last-child')
        mousedown(column)

      it 'changes the sort order to use the clicked column', ->
