 
       runs ->
         expect(editorView.rootElement.querySelectorAll('.git-line-modified').length).toBe 1
-        expect(editorView.rootElement.querySelector('.git-line-modified')).toHaveClass('line-number-0')
+        expect(editorView.rootElement.querySelector('.git-line-modified')).toHaveData("buffer-row", 0)
 
   describe "when the project paths change", ->
     it "doesn't try to use the destroyed git repository", ->

      editor.deleteLine()
      advanceClock(editor.getBuffer().stoppedChangingDelay)
      expect(editorView.rootElement.querySelectorAll('.git-line-removed').length).toBe 1
      expect(editorView.rootElement.querySelector('.git-line-removed')).toHaveClass('line-number-4')

  describe "when a modified line is restored to the HEAD version contents", ->
    it "removes the diff highlight", ->

      editor.insertText('a')
      advanceClock(editor.getBuffer().stoppedChangingDelay)
      expect(editorView.rootElement.querySelectorAll('.git-line-modified').length).toBe 1
      expect(editorView.rootElement.querySelector('.git-line-modified')).toHaveData("buffer-row", 0)

  describe "when the editor has added lines", ->
    it "highlights the added lines", ->
