diff --git a/src/text-editor-component.coffee b/src/text-editor-component.coffee
index d43e499b3..8b0f71a4c 100644
--- a/src/text-editor-component.coffee
+++ b/src/text-editor-component.coffee
@@ -908,10 +908,10 @@ TextEditorComponent = React.createClass
   lineNumberNodeForScreenRow: (screenRow) -> @refs.gutter.lineNumberNodeForScreenRow(screenRow)
 
   screenRowForNode: (node) ->
-    while node isnt document
+    while node?
       if screenRow = node.dataset.screenRow
         return parseInt(screenRow)
-      node = node.parentNode
+      node = node.parentElement
     null
 
   getFontSize: ->