From 685f599c822e9d2b6b51c37545399ff271e06614 Mon Sep 17 00:00:00 2001
From: Ivan Kylchik <ivan.kylchik@jetbrains.com>
Date: Sat, 20 Feb 2021 15:38:08 +0300
Subject: [PATCH] [VISUALIZER] Allow to render all calls at unary expresion in
 fir

---
 .../org/jetbrains/kotlin/compiler/visualizer/FirVisualizer.kt   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/compiler/visualizer/render-fir/src/org/jetbrains/kotlin/compiler/visualizer/FirVisualizer.kt b/compiler/visualizer/render-fir/src/org/jetbrains/kotlin/compiler/visualizer/FirVisualizer.kt
index c617224da38..13aff49e35f 100644
--- a/compiler/visualizer/render-fir/src/org/jetbrains/kotlin/compiler/visualizer/FirVisualizer.kt
+++ b/compiler/visualizer/render-fir/src/org/jetbrains/kotlin/compiler/visualizer/FirVisualizer.kt
@@ -215,7 +215,7 @@ class FirVisualizer(private val firFile: FirFile) : BaseRenderer() {
                 expression.baseExpression?.accept(this)
                 return
             }
-            expression.firstOfTypeWithLocalReplace<FirFunctionCall>(expression.operationReference) { this.calleeReference.name.asString() }
+            expression.allOfTypeWithLocalReplace<FirFunctionCall>(expression.operationReference) { this.calleeReference.name.asString() }
             super.visitUnaryExpression(expression)
         }