diff --git a/idea/src/org/jetbrains/kotlin/idea/formatter/TrailingCommaVisitor.kt b/idea/src/org/jetbrains/kotlin/idea/formatter/TrailingCommaVisitor.kt
index 904724c63fe..34cb42e2e04 100644
--- a/idea/src/org/jetbrains/kotlin/idea/formatter/TrailingCommaVisitor.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/formatter/TrailingCommaVisitor.kt
@@ -51,10 +51,9 @@ abstract class TrailingCommaVisitor : KtTreeVisitorVoid() {
     }
 
     override fun visitElement(element: PsiElement) {
-        if (recursively)
-            super.visitElement(element)
-        else
-            ProgressIndicatorProvider.checkCanceled()
+        ProgressIndicatorProvider.checkCanceled()
+
+        if (recursively) super.visitElement(element)
     }
 
     protected abstract fun process(commaOwner: KtElement)