diff --git a/idea/idea-completion/testData/handlers/keywords/ReturnInEmptyType.kt b/idea/idea-completion/testData/handlers/keywords/ReturnInEmptyType.kt
index caf3aaaddc2..7a397ef704d 100644
--- a/idea/idea-completion/testData/handlers/keywords/ReturnInEmptyType.kt
+++ b/idea/idea-completion/testData/handlers/keywords/ReturnInEmptyType.kt
@@ -1,3 +1,4 @@
+// ELEMENT: return
 fun test(a: Int) {
     retur<caret>
-}
\ No newline at end of file
+}
diff --git a/idea/idea-completion/testData/handlers/keywords/ReturnInEmptyType.kt.after b/idea/idea-completion/testData/handlers/keywords/ReturnInEmptyType.kt.after
index 2c31061fbf9..751ed46b2bd 100644
--- a/idea/idea-completion/testData/handlers/keywords/ReturnInEmptyType.kt.after
+++ b/idea/idea-completion/testData/handlers/keywords/ReturnInEmptyType.kt.after
@@ -1,3 +1,4 @@
+// ELEMENT: return
 fun test(a: Int) {
     return<caret>
-}
\ No newline at end of file
+}
diff --git a/idea/idea-completion/testData/handlers/keywords/ReturnInProperty.kt b/idea/idea-completion/testData/handlers/keywords/ReturnInProperty.kt
index be4a18be817..5ffe44012a7 100644
--- a/idea/idea-completion/testData/handlers/keywords/ReturnInProperty.kt
+++ b/idea/idea-completion/testData/handlers/keywords/ReturnInProperty.kt
@@ -1,6 +1,7 @@
+// ELEMENT: return
 class Test {
     val test : String
         get() {
             retur<caret>
         }
-}
\ No newline at end of file
+}
diff --git a/idea/idea-completion/testData/handlers/keywords/ReturnInProperty.kt.after b/idea/idea-completion/testData/handlers/keywords/ReturnInProperty.kt.after
index a3f11d43a2f..1b358c88a8a 100644
--- a/idea/idea-completion/testData/handlers/keywords/ReturnInProperty.kt.after
+++ b/idea/idea-completion/testData/handlers/keywords/ReturnInProperty.kt.after
@@ -1,6 +1,7 @@
+// ELEMENT: return
 class Test {
     val test : String
         get() {
             return <caret>
         }
-}
\ No newline at end of file
+}
diff --git a/idea/idea-completion/testData/handlers/keywords/ReturnInTypeFunction.kt b/idea/idea-completion/testData/handlers/keywords/ReturnInTypeFunction.kt
index d060ce9b037..4b0fbe384fa 100644
--- a/idea/idea-completion/testData/handlers/keywords/ReturnInTypeFunction.kt
+++ b/idea/idea-completion/testData/handlers/keywords/ReturnInTypeFunction.kt
@@ -1,5 +1,6 @@
+// ELEMENT: return
 class Test {
     fun someMethod() : Int {
         retur<caret>
     }
-}
\ No newline at end of file
+}
diff --git a/idea/idea-completion/testData/handlers/keywords/ReturnInTypeFunction.kt.after b/idea/idea-completion/testData/handlers/keywords/ReturnInTypeFunction.kt.after
index e05c5eb5fe9..4c539e9152a 100644
--- a/idea/idea-completion/testData/handlers/keywords/ReturnInTypeFunction.kt.after
+++ b/idea/idea-completion/testData/handlers/keywords/ReturnInTypeFunction.kt.after
@@ -1,5 +1,6 @@
+// ELEMENT: return
 class Test {
     fun someMethod() : Int {
         return <caret>
     }
-}
\ No newline at end of file
+}
diff --git a/idea/idea-completion/testData/handlers/keywords/ReturnInUnit.kt b/idea/idea-completion/testData/handlers/keywords/ReturnInUnit.kt
index 87c304b1538..90d29b88d12 100644
--- a/idea/idea-completion/testData/handlers/keywords/ReturnInUnit.kt
+++ b/idea/idea-completion/testData/handlers/keywords/ReturnInUnit.kt
@@ -1,5 +1,6 @@
+// ELEMENT: return
 object OtherTest {
     fun test() : Unit {
         retur<caret>
     }
-}
\ No newline at end of file
+}
diff --git a/idea/idea-completion/testData/handlers/keywords/ReturnInUnit.kt.after b/idea/idea-completion/testData/handlers/keywords/ReturnInUnit.kt.after
index 044a851ecb6..a6b9ee1e2f7 100644
--- a/idea/idea-completion/testData/handlers/keywords/ReturnInUnit.kt.after
+++ b/idea/idea-completion/testData/handlers/keywords/ReturnInUnit.kt.after
@@ -1,5 +1,6 @@
+// ELEMENT: return
 object OtherTest {
     fun test() : Unit {
         return<caret>
     }
-}
\ No newline at end of file
+}