diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetModifiableBlockHelper.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetModifiableBlockHelper.java
index 9fccc4574c5..e78f20432b1 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetModifiableBlockHelper.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetModifiableBlockHelper.java
@@ -27,7 +27,6 @@ public final class JetModifiableBlockHelper {
     }
 
     // TODO: Need tests for this method
-    // TODO: Consider for FUNCTION_LITERALS
     public static boolean shouldChangeModificationCount(PsiElement place) {
         JetDeclaration declaration = PsiTreeUtil.getParentOfType(place, JetDeclaration.class, true);
         if (declaration != null) {
@@ -47,6 +46,9 @@ public final class JetModifiableBlockHelper {
 
                 return shouldChangeModificationCount(property);
             }
+            else if (declaration instanceof JetFunctionLiteral) {
+                return shouldChangeModificationCount(declaration);
+            }
         }
 
         return true;