diff --git a/compiler/testData/diagnostics/tests/inference/extensionLambdasAndArrow.fir.kt b/compiler/testData/diagnostics/tests/inference/extensionLambdasAndArrow.fir.kt
index 4a7820d4273..1c13bf01c69 100644
--- a/compiler/testData/diagnostics/tests/inference/extensionLambdasAndArrow.fir.kt
+++ b/compiler/testData/diagnostics/tests/inference/extensionLambdasAndArrow.fir.kt
@@ -1,5 +1,7 @@
 // !DIAGNOSTICS: -UNUSED_VARIABLE -UNUSED_ANONYMOUS_PARAMETER
 
+fun <T> select(vararg x: T) = x[0]
+
 fun main() {
     val x1: String.() -> String = if (true) {{ this }} else {{ this }}
     val x2: String.() -> String = if (true) {{ -> this }} else {{ -> this }}
@@ -7,4 +9,6 @@ fun main() {
     val x4: String.() -> String = if (true) {{ str: String -> "this" }} else {{ str: String -> "this" }}
     val x5: String.() -> String = if (true) {{ str -> "this" }} else {{ str -> "this" }}
     val x6: String.() -> String = if (true) {{ str -> "this" }} else {{ "this" }}
+    val x7: String.() -> String = select({ -> this }, { -> this })
+    val x8: String.() -> String = select({ this }, { this })
 }