diff --git a/buildSrc/src/main/kotlin/idea/DistModelBuilder.kt b/buildSrc/src/main/kotlin/idea/DistModelBuilder.kt
index 32dd982683b..0c313e494a0 100755
--- a/buildSrc/src/main/kotlin/idea/DistModelBuilder.kt
+++ b/buildSrc/src/main/kotlin/idea/DistModelBuilder.kt
@@ -180,14 +180,14 @@ open class DistModelBuilder(val rootProject: Project, pw: PrintWriter) {
                 })
             }
             sourcePath is CompositeFileCollection -> ctx.child("COMPOSITE FILE COLLECTION") { child ->
-                sourcePath.visitRootElements(object : FileCollectionVisitor {
+                sourcePath.visitLeafCollections(object : FileCollectionLeafVisitor {
                     override fun visitDirectoryTree(directoryTree: DirectoryFileTree) {
                         child.child("DIR TREE") {
                             it.addCopyOf(directoryTree.dir.path)
                         }
                     }
 
-                    override fun visitTree(fileTree: FileTreeInternal) {
+                    override fun visitGenericFileTree(fileTree: FileTreeInternal) {
                         child.child("TREE") {
                             processSourcePath(fileTree, it)
                         }