diff --git a/compiler/fir/tree/build.gradle.kts b/compiler/fir/tree/build.gradle.kts
index eb99997f2c0..b4752f7b8ea 100644
--- a/compiler/fir/tree/build.gradle.kts
+++ b/compiler/fir/tree/build.gradle.kts
@@ -29,6 +29,14 @@ dependencies {
 
 val generateTree by tasks.registering(NoDebugJavaExec::class) {
     val generationRoot = "$projectDir/src/"
+    val generatorRoot = "$projectDir/tree-generator/src/"
+
+    val generatorConfigurationFiles = fileTree(generatorRoot) {
+        include("**/*.kt")
+    }
+
+    inputs.files(generatorConfigurationFiles)
+    outputs.dirs(generationRoot)
 
     args(generationRoot)
     classpath = generatorClasspath