diff --git a/idea/testData/checker/FunctionReturnTypes.kt b/idea/testData/checker/FunctionReturnTypes.kt
index 9ccd9e1288a..49e458a892a 100644
--- a/idea/testData/checker/FunctionReturnTypes.kt
+++ b/idea/testData/checker/FunctionReturnTypes.kt
@@ -6,7 +6,7 @@ fun unitEmptyReturn() : Unit {return}
 fun unitIntReturn() : Unit {return <error>1</error>}
 fun unitUnitReturn() : Unit {return Unit.VALUE}
 fun test1() : Any = { <error>return</error> }
-fun test2() : Any = @a {return@a 1}
+fun test2() : Any = @a {<error>return@a 1</error>}
 fun test3() : Any { <error>return</error> }
 
 fun bbb() {