diff --git a/compiler/frontend/src/org/jetbrains/kotlin/types/expressions/FunctionsTypingVisitor.kt b/compiler/frontend/src/org/jetbrains/kotlin/types/expressions/FunctionsTypingVisitor.kt
index 478c3040aa5..a182279564d 100644
--- a/compiler/frontend/src/org/jetbrains/kotlin/types/expressions/FunctionsTypingVisitor.kt
+++ b/compiler/frontend/src/org/jetbrains/kotlin/types/expressions/FunctionsTypingVisitor.kt
@@ -99,10 +99,10 @@ public class FunctionsTypingVisitor(facade: ExpressionTypingInternals) : Express
         }
 
         if (isStatement) {
-            return DataFlowUtils.checkStatementType(function, context as ResolutionContext<*>, context.dataFlowInfo)
+            return DataFlowUtils.checkStatementType(function, context, context.dataFlowInfo)
         }
         else {
-            return DataFlowUtils.checkType(createFunctionType(functionDescriptor), function, context as ResolutionContext<*>, context.dataFlowInfo)
+            return DataFlowUtils.checkType(createFunctionType(functionDescriptor), function, context, context.dataFlowInfo)
         }
     }
 
@@ -137,7 +137,7 @@ public class FunctionsTypingVisitor(facade: ExpressionTypingInternals) : Express
             return JetTypeInfo.create(resultType, context.dataFlowInfo)
         }
 
-        return DataFlowUtils.checkType(resultType, expression, context as ResolutionContext<*>, context.dataFlowInfo)
+        return DataFlowUtils.checkType(resultType, expression, context, context.dataFlowInfo)
     }
 
     private fun createFunctionDescriptor(