From 04d858474252b1b71b313e601ee8f4d000fb9840 Mon Sep 17 00:00:00 2001
From: Kevin Sawicki <kevin@github.com>
Date: Tue, 11 Feb 2014 13:16:58 -0800
Subject: [PATCH] Use fold directly

---
 src/editor.coffee | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/editor.coffee b/src/editor.coffee
index 693f0d675..c9e15a883 100644
--- a/src/editor.coffee
+++ b/src/editor.coffee
@@ -694,11 +694,10 @@ class Editor extends Model
         insertDelta = 1
 
       for row in rows
-        screenRow = @screenPositionForBufferPosition([row]).row
-        if @isFoldedAtScreenRow(screenRow)
-          bufferRange = @bufferRangeForScreenRange([[screenRow], [screenRow + 1]])
+        if fold = @displayBuffer.largestFoldStartingAtBufferRow(row)
+          bufferRange = fold.getBufferRange()
           startRow = bufferRange.start.row
-          endRow = bufferRange.end.row - 1
+          endRow = bufferRange.end.row
           foldedRows.push(endRow + insertDelta)
         else
           startRow = row