[FIR PSI] Set source of equality operator call

#KT-55835
2023-06-28 15:19:30 +03:00
parent 9e9b92fd16
commit 95005dae8a
1 changed files with 1 additions and 1 deletions
@@ -36,7 +36,7 @@ internal fun KtWhenCondition.toFirWhenCondition(
    return when (this) {
        is KtWhenConditionWithExpression -> {
            buildEqualityOperatorCall {
-                source = expression?.toKtPsiSourceElement(KtFakeSourceElementKind.WhenCondition)
+                source = (expression ?: firstChild)?.toKtPsiSourceElement(KtFakeSourceElementKind.WhenCondition)
                operation = FirOperation.EQ
                argumentList = buildBinaryArgumentList(
                    firSubjectExpression, expression.convert("No expression in condition with expression")