diff --git a/idea/testData/fir/lazyResolve/elvis/main.txt b/idea/testData/fir/lazyResolve/elvis/main.txt
index 557c9ec61c9..416b24304ca 100644
--- a/idea/testData/fir/lazyResolve/elvis/main.txt
+++ b/idea/testData/fir/lazyResolve/elvis/main.txt
@@ -8,15 +8,7 @@ FILE: main.kt
             public get(): R|kotlin/String?|
 
         public final fun foo(): R|kotlin/String| {
-            ^foo when (lval <elvis>: R|kotlin/String?| = this@R|elvis/WithElvis|.R|elvis/WithElvis.value|) {
-                ==($subj$, Null(null)) ->  {
-                    String()
-                }
-                else ->  {
-                    R|<local>/<elvis>|
-                }
-            }
-
+            ^foo this@R|elvis/WithElvis|.R|elvis/WithElvis.value| ?: String()
         }
 
     }