           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 "adds an underline to the current line", ->
          expect(atom.workspaceView
            .find('.cursor-line.highlight-line-multi-line-dotted-bottom'))
            .toHaveLength(1)

    describe "when there are two cursors", ->
      beforeEach ->

          editor.setSelectedBufferRange(range)

        it "adds an underline to the current line", ->
          expect(
            editorElement.shadowRoot.querySelectorAll(
              '.cursor-line.highlight-line-multi-line-dashed-bottom'
            )
          ).toHaveLength(1)

      describe "when dotted settings has been set", ->
        beforeEach ->
