diff --git a/compiler/tests-common-new/tests/org/jetbrains/kotlin/test/frontend/fir/handlers/FirDiagnosticsHandler.kt b/compiler/tests-common-new/tests/org/jetbrains/kotlin/test/frontend/fir/handlers/FirDiagnosticsHandler.kt index 9065795c12d..1f7045b0fed 100644 --- a/compiler/tests-common-new/tests/org/jetbrains/kotlin/test/frontend/fir/handlers/FirDiagnosticsHandler.kt +++ b/compiler/tests-common-new/tests/org/jetbrains/kotlin/test/frontend/fir/handlers/FirDiagnosticsHandler.kt @@ -413,8 +413,7 @@ private val KtSourceElement.operatorSignIfBinary: KtSourceElement? get() = when (elementType) { KtNodeTypes.BINARY_EXPRESSION -> when (this) { is KtPsiSourceElement -> (psi as? KtBinaryExpression)?.operationReference?.toKtPsiSourceElement(kind) - is KtLightSourceElement -> treeStructure.getParent(lighterASTNode) - ?.let { treeStructure.findChildByType(it, KtNodeTypes.OPERATION_REFERENCE) } + is KtLightSourceElement -> treeStructure.findChildByType(lighterASTNode, KtNodeTypes.OPERATION_REFERENCE) ?.toKtLightSourceElement(treeStructure, kind) else -> null }