diff --git a/lib/ace/layer/text.js b/lib/ace/layer/text.js
index eac5dc0e..aabefabd 100644
--- a/lib/ace/layer/text.js
+++ b/lib/ace/layer/text.js
@@ -191,10 +191,7 @@ var Text = function(parentEl) {
 
         for (var row = config.firstRow; row < first; row++) {
             lineElementsIdx ++;
-            if (this.session.isRowFolded(row)) {
-                var fold = this.session.getRowLastFold(row);
-                row = fold.end.row;
-            }
+            row = this.session.getRowFoldEnd(row);
         }
 
         for (var i=first; i<=last; i++) {