diff --git a/compiler/frontend/src/org/jetbrains/kotlin/types/expressions/ExpressionTypingServices.java b/compiler/frontend/src/org/jetbrains/kotlin/types/expressions/ExpressionTypingServices.java
index 788642efaae..01f721c29d2 100644
--- a/compiler/frontend/src/org/jetbrains/kotlin/types/expressions/ExpressionTypingServices.java
+++ b/compiler/frontend/src/org/jetbrains/kotlin/types/expressions/ExpressionTypingServices.java
@@ -297,7 +297,7 @@ public class ExpressionTypingServices {
                 beforeJumpInfo = result.getJumpFlowInfo();
                 jumpOutPossible = result.getJumpOutPossible();
             }
-            if (newDataFlowInfo != context.dataFlowInfo) {
+            if (newDataFlowInfo != newContext.dataFlowInfo) {
                 newContext = newContext.replaceDataFlowInfo(newDataFlowInfo);
                 // We take current data flow info if jump there is not possible
             }