           editor.setSelectedBufferRange(range)
 
         it "adds an underline to the current line", ->
-          expect(atom.workspaceView
-            .find('.cursor-line.highlight-line-multi-line-dashed-bottom'))
-            .toHaveLength(1)
+          expect(
+            editorElement.shadowRoot.querySelectorAll(
+              '.cursor-line.highlight-line-multi-line-dashed-bottom'
+            )
+          ).toHaveLength(1)
 
       describe "when dotted settings has been set", ->
         beforeEach ->


  describe "when the view is loaded", ->
    it "attaches the view", ->
      expect(atom.workspaceView.find('.highlight-line')).toExist()

  describe "when the background color is enabled", ->
    beforeEach ->

        editor.setSelectedBufferRange(range)

      it "won`t add a highlight line class", ->
        expect(editorElement.shadowRoot
          .querySelectorAll('.cursor-line.highlight-line')
        ).toHaveLength(0)

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