diff --git a/gradle.properties b/gradle.properties
index 364a910..89b3936 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -1,3 +1,3 @@
 kotlin.code.style=official
-loom_version=0.6.33
+loom_version=0.6.54
 base_version=2.0
\ No newline at end of file
diff --git a/src/main/kotlin/me/shedaniel/architect/plugin/ArchitectPluginExtension.kt b/src/main/kotlin/me/shedaniel/architect/plugin/ArchitectPluginExtension.kt
index 8a6ffa6..8ec19cb 100644
--- a/src/main/kotlin/me/shedaniel/architect/plugin/ArchitectPluginExtension.kt
+++ b/src/main/kotlin/me/shedaniel/architect/plugin/ArchitectPluginExtension.kt
@@ -21,6 +21,7 @@ open class ArchitectPluginExtension(val project: Project) {
     fun platformSetupLoomIde() {
         val loomExtension = project.extensions.getByType(LoomGradleExtension::class.java)
         loomExtension.autoGenIDERuns = true
+        loomExtension.addTaskBeforeRun("\$PROJECT_DIR\$/${project.name}:classes")
     }
 
     fun common(forgeEnabled: Boolean) {
diff --git a/src/main/kotlin/me/shedaniel/architect/plugin/TransformingTask.kt b/src/main/kotlin/me/shedaniel/architect/plugin/TransformingTask.kt
index db2382f..8500774 100644
--- a/src/main/kotlin/me/shedaniel/architect/plugin/TransformingTask.kt
+++ b/src/main/kotlin/me/shedaniel/architect/plugin/TransformingTask.kt
@@ -3,6 +3,8 @@ package me.shedaniel.architect.plugin
 import me.shedaniel.architect.plugin.utils.GradleSupport
 import org.gradle.api.Project
 import org.gradle.api.file.RegularFileProperty
+import org.gradle.api.tasks.Input
+import org.gradle.api.tasks.InputFile
 import org.gradle.api.tasks.TaskAction
 import org.gradle.jvm.tasks.Jar
 import java.io.File
@@ -16,7 +18,9 @@ import kotlin.time.ExperimentalTime
 import kotlin.time.nanoseconds
 
 open class TransformingTask : Jar() {
+    @InputFile
     val input: RegularFileProperty = GradleSupport.getFileProperty(project)
+    @Input
     val transformers = mutableListOf<Transformer>()
 
     @ExperimentalTime