           it "will still have a line", ->
             range = new Range(new Point(8, 2), new Point(8, 5))
             editor.setSelectedBufferRange(range)
-            expect(atom.workspaceView
-              .find('.line.highlight-line-multi-line-solid-bottom'))
-              .toHaveLength(1)
+            expect(
+              editorElement.shadowRoot.querySelectorAll(
+                '.line.highlight-line-multi-line-solid-bottom'
+              )
+            ).toHaveLength(1)
 
       describe "when dashed settings has been set", ->
         beforeEach ->

        editor.setSelectedBufferRange(range)

      it "won`t add a highlight line class", ->
        expect(atom.workspaceView.find('.cursor-line.highlight-line'))
          .toHaveLength(0)

      describe "when selection border is enabled", ->
        beforeEach ->

        editor.setSelectedBufferRange(range)

      it "adds the background class to the cursor line", ->
        expect(editorElement.shadowRoot
          .querySelectorAll('.cursor-line.highlight-line')
        ).toHaveLength(1)

      describe "when hide highlight on select is enabled", ->
        beforeEach ->
          atom.config.set('highlight-line.hideHighlightOnSelect', true)

        it "will have a highlight when there is no text selected", ->
          expect(editorElement.shadowRoot
            .querySelectorAll('.cursor-line.highlight-line')
          ).toHaveLength(1)

        it "won`t have a highlight when there is text selected", ->
          range = new Range(new Point(8, 2), new Point(8, 5))
          editor.setSelectedBufferRange(range)
          expect(editorElement.shadowRoot
            .querySelectorAll('.cursor-line.highlight-line')
          ).toHaveLength(0)

    describe "when underline is enabled", ->
      beforeEach ->
