From cfb2d39c7868263d11d4903d10e918a2e6d3c5e5 Mon Sep 17 00:00:00 2001
From: Dmitry Batrak <Dmitry.Batrak@jetbrains.com>
Date: Fri, 15 Aug 2014 19:41:06 +0400
Subject: [PATCH] EA-57803 - correcting previous fix for injected folding pass
 (EA-54811)

---
 .../com/intellij/codeInsight/folding/impl/FoldingUpdate.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/platform/lang-impl/src/com/intellij/codeInsight/folding/impl/FoldingUpdate.java b/platform/lang-impl/src/com/intellij/codeInsight/folding/impl/FoldingUpdate.java
index 7d0b4453c517..4c137f8d2a37 100644
--- a/platform/lang-impl/src/com/intellij/codeInsight/folding/impl/FoldingUpdate.java
+++ b/platform/lang-impl/src/com/intellij/codeInsight/folding/impl/FoldingUpdate.java
@@ -174,7 +174,7 @@ public class FoldingUpdate {
           injectedFiles.add(injectedFile);
           final FoldingMap map = new FoldingMap();
           maps.add(map);
-          getFoldingsFor(injectedFile, injectedDocument, map, false);
+          getFoldingsFor(injectedFile, injectedEditor.getDocument(), map, false);
         }
       });
     }