Partial body resolve: corrected it for explicit function type but unfortunately it does not work for non-explicit type yet

idea/ide-common/src/org/jetbrains/jet/lang/resolve/lazy/PartialBodyResolveFilter.kt

@@ -68,6 +68,10 @@ class PartialBodyResolveFilter(
     //TODO: do..while is special case
 
     private fun processBlock(block: JetBlockExpression): NameFilter {
+        if (isValueNeeded(block)) {
+            block.lastStatement()?.let { statementTree.mark(it, MarkLevel.NEED_REFERENCE_RESOLVE) }
+        }
+
         val nameFilter = NameFilter()
         if (!statementTree.hasMarks(block, MarkLevel.NEED_REFERENCE_RESOLVE)) return nameFilter
 
@@ -117,9 +121,6 @@ class PartialBodyResolveFilter(
             val level = statementTree.statementMark(statement)
             if (level > MarkLevel.RESOLVE_STATEMENT) {
                 for (nestedBlock in statementTree.blocks(statement)) {
-                    if (isValueNeeded(nestedBlock)) {
-                        nestedBlock.lastStatement()?.let { statementTree.mark(it, MarkLevel.NEED_REFERENCE_RESOLVE) }
-                    }
                     val childFilter = processBlock(nestedBlock)
                     nameFilter.addNames(childFilter)
                 }