diff --git a/compiler/testData/diagnostics/tests/inference/kt6175.fir.kt b/compiler/testData/diagnostics/tests/inference/kt6175.fir.kt
index 75ccd5a614d..ed4b4a4b090 100644
--- a/compiler/testData/diagnostics/tests/inference/kt6175.fir.kt
+++ b/compiler/testData/diagnostics/tests/inference/kt6175.fir.kt
@@ -5,10 +5,10 @@ fun <T, R : Any> foo(body: (R?) -> T): T = fail()
 
 fun test1() {
     foo {
-        true
+        <!ARGUMENT_TYPE_MISMATCH!>true<!>
     }
     foo { x ->
-        true
+        <!ARGUMENT_TYPE_MISMATCH!>true<!>
     }
 }
 
@@ -17,10 +17,10 @@ fun <T, R> bar(body: (R) -> T): T = fail()
 
 fun test2() {
     bar {
-        true
+        <!ARGUMENT_TYPE_MISMATCH!>true<!>
     }
     bar { x ->
-        true
+        <!ARGUMENT_TYPE_MISMATCH!>true<!>
     }
 }
 
@@ -28,10 +28,10 @@ fun <T, R> baz(body: (List<R>) -> T): T = fail()
 
 fun test3() {
     baz {
-        true
+        <!ARGUMENT_TYPE_MISMATCH!>true<!>
     }
     baz { x ->
-        true
+        <!ARGUMENT_TYPE_MISMATCH!>true<!>
     }
 }
 
@@ -39,10 +39,10 @@ fun <T, R : Any> brr(body: (List<R?>) -> T): T = fail()
 
 fun test4() {
     brr {
-        true
+        <!ARGUMENT_TYPE_MISMATCH!>true<!>
     }
     brr { x ->
-        true
+        <!ARGUMENT_TYPE_MISMATCH!>true<!>
     }
 }
 
diff --git a/compiler/testData/diagnostics/tests/inference/reportingImprovements/cannotInferParameterTypeWithInference.fir.kt b/compiler/testData/diagnostics/tests/inference/reportingImprovements/cannotInferParameterTypeWithInference.fir.kt
index 04f0dd15e3b..80e64e539ee 100644
--- a/compiler/testData/diagnostics/tests/inference/reportingImprovements/cannotInferParameterTypeWithInference.fir.kt
+++ b/compiler/testData/diagnostics/tests/inference/reportingImprovements/cannotInferParameterTypeWithInference.fir.kt
@@ -6,7 +6,7 @@ fun <T, R> foo(block: (T)-> R) = block
 fun test1() {
     foo {
         x ->  // here we have 'cannot infer parameter type' error
-        43
+        <!ARGUMENT_TYPE_MISMATCH!>43<!>
     }
 }
 
diff --git a/compiler/testData/diagnostics/tests/smartCasts/lambdaArgumentWithBoundWithoutType.fir.kt b/compiler/testData/diagnostics/tests/smartCasts/lambdaArgumentWithBoundWithoutType.fir.kt
index 2dbb729062c..ab39cb96649 100644
--- a/compiler/testData/diagnostics/tests/smartCasts/lambdaArgumentWithBoundWithoutType.fir.kt
+++ b/compiler/testData/diagnostics/tests/smartCasts/lambdaArgumentWithBoundWithoutType.fir.kt
@@ -16,11 +16,11 @@ val foo: Foo = run {
 val foofoo: Foo = run {
     val x = foo()
     if (x == null) throw Exception()
-    x
+    <!ARGUMENT_TYPE_MISMATCH!>x<!>
 }
 
 val bar: Bar = run {
     val x = foo()
     if (x == null) throw Exception()
-    x
+    <!ARGUMENT_TYPE_MISMATCH!>x<!>
 }