diff --git a/compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/dataFlowInfoAfterExclExcl.fir.kt b/compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/dataFlowInfoAfterExclExcl.fir.kt
index c533df109b8..f010cf53a13 100644
--- a/compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/dataFlowInfoAfterExclExcl.fir.kt
+++ b/compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/dataFlowInfoAfterExclExcl.fir.kt
@@ -1,3 +1,5 @@
+// !WITH_NEW_INFERENCE
+
 fun foo(d: Any?) {
     if (d is String?) {
         d!!
diff --git a/compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/dataFlowInfoAfterExclExcl.kt b/compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/dataFlowInfoAfterExclExcl.kt
index 97d793a3369..2e8b39b2dd5 100644
--- a/compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/dataFlowInfoAfterExclExcl.kt
+++ b/compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/dataFlowInfoAfterExclExcl.kt
@@ -1,7 +1,9 @@
+// !WITH_NEW_INFERENCE
+
 fun foo(d: Any?) {
     if (d is String?) {
         <!DEBUG_INFO_SMARTCAST!>d<!>!!
-        doString(<!DEBUG_INFO_SMARTCAST!>d<!>)
+        doString(<!NI;TYPE_MISMATCH, OI;DEBUG_INFO_SMARTCAST!>d<!>)
     }
 }