diff --git a/compiler/fir/resolve/testData/resolve/whenExpressionType.kt b/compiler/fir/resolve/testData/resolve/whenExpressionType.kt
index c8343a057c5..3b4917099dc 100644
--- a/compiler/fir/resolve/testData/resolve/whenExpressionType.kt
+++ b/compiler/fir/resolve/testData/resolve/whenExpressionType.kt
@@ -38,7 +38,7 @@ fun case3() {
         A.A1 -> B()
         A.A2 -> B()
     }
-    val x1 = <!INAPPLICABLE_CANDIDATE!>C<!>(l1) //INAPPLICABLE_CANDIDATE (l1 is Unit)
+    val x1 = C(l1)
 }
 
 // TESTCASE NUMBER: 4
diff --git a/compiler/fir/resolve/testData/resolve/whenExpressionType.txt b/compiler/fir/resolve/testData/resolve/whenExpressionType.txt
index 1d8f558fa6a..e08136acf00 100644
--- a/compiler/fir/resolve/testData/resolve/whenExpressionType.txt
+++ b/compiler/fir/resolve/testData/resolve/whenExpressionType.txt
@@ -56,7 +56,7 @@ FILE: whenExpressionType.kt
     }
     public final fun case3(): R|kotlin/Unit| {
         lval flag: R|A| = Q|A|.R|/A.A1|
-        lval l1: R|kotlin/Unit| = when (R|<local>/flag|!!) {
+        lval l1: R|B| = when (R|<local>/flag|!!) {
             ==($subj$, Q|A|.R|/A.A1|) ->  {
                 R|/B.B|()
             }
@@ -65,7 +65,7 @@ FILE: whenExpressionType.kt
             }
         }
 
-        lval x1: <ERROR TYPE REF: Inapplicable(INAPPLICABLE): [/C.C]> = <Inapplicable(INAPPLICABLE): [/C.C]>#(R|<local>/l1|)
+        lval x1: R|C| = R|/C.C|(R|<local>/l1|)
     }
     public final fun case4(): R|kotlin/Unit| {
         lval flag: R|A| = Q|A|.R|/A.A1|