Fix tests in DiagnosticsTestWithStdLibGenerated with NI

compiler/testData/diagnostics/testsWithStdLib/when/noTypeArgumentsInConstructor.kt

@@ -1,3 +1,5 @@
+// !WITH_NEW_INFERENCE
+
 import java.util.*
 
 fun <T> nullable(x: T): T? = x
@@ -26,7 +28,7 @@ val test4: Collection<Int> =
         }
 
 val test5: Collection<Int> =
-        listOf(1, 2, 3).<!TYPE_INFERENCE_EXPECTED_TYPE_MISMATCH!>flatMapTo(<!TYPE_INFERENCE_NO_INFORMATION_FOR_PARAMETER!>LinkedHashSet<!>()) { // TODO
+        listOf(1, 2, 3).<!OI;TYPE_INFERENCE_EXPECTED_TYPE_MISMATCH!>flatMapTo(<!OI;TYPE_INFERENCE_NO_INFORMATION_FOR_PARAMETER!>LinkedHashSet<!>()) { // TODO
             if (true) listOf(it) else listOf(it)
         }<!>