       symbol = /'(?:[^.]+\.)?([^']+)'/.exec(message)[1]
       while true
         offset = 0
-        tokenizedLine = textEditor.tokenizedLineForScreenRow(lineNumber)
+        tokenizedLine = tokenizedLineForRow(textEditor, lineNumber)
         if tokenizedLine is undefined
           break
         foundDecorator = false

extractRange = ({code, message, lineNumber, colNumber, textEditor}) ->
  switch code
    when 'C901'

      symbol = /'([^']+)'/.exec(message)[1]
      while true
        offset = 0
        tokenizedLine = tokenizedLineForRow(textEditor, lineNumber)
        if tokenizedLine is undefined
          break
        for token in tokenizedLine.tokens
