       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'

        offset += token.bufferDelta
    when 'W291'
      # W291 - trailing whitespace
      screenLine = tokenizedLineForRow(textEditor, lineNumber)
      if screenLine is undefined
        break
      return [[lineNumber, colNumber - 1], [lineNumber, screenLine.length]]
