diff --git a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/caches/ValueWithPostCompute.kt b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/caches/ValueWithPostCompute.kt
index 35c20c8b095..c6694765d15 100644
--- a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/caches/ValueWithPostCompute.kt
+++ b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/caches/ValueWithPostCompute.kt
@@ -92,7 +92,7 @@ internal class ValueWithPostCompute<KEY, VALUE, DATA>(
                 throw stateSnapshot.error
             }
             else -> {
-                return value as VALUE
+                return stateSnapshot as VALUE
             }
         }
     }