diff --git a/src/main/java/net/legacyfabric/legacylooming/LegacyLoomingGradlePlugin.java b/src/main/java/net/legacyfabric/legacylooming/LegacyLoomingGradlePlugin.java
index ff914a6..2693e55 100644
--- a/src/main/java/net/legacyfabric/legacylooming/LegacyLoomingGradlePlugin.java
+++ b/src/main/java/net/legacyfabric/legacylooming/LegacyLoomingGradlePlugin.java
@@ -79,8 +79,10 @@ private static void setManifestAttributes(Project project, LegacyLoomingExtensio
 
     public static void overrideMigrateTask(Project project) {
         // override loom's migrateMappings to fix issues
-        var migrateTask = project.getTasks().replace("improvedMigrateMappings", MigrateLegacyMappingsTask.class);
-        migrateTask.setDescription("Migrates mappings to a new version.");
-        migrateTask.getOutputs().upToDateWhen(o -> false);
+        project.getTasks().register("improvedMigrateMappings", MigrateLegacyMappingsTask.class, migrateTask -> {
+            migrateTask.setDescription("Migrates mappings to a new version.");
+            migrateTask.getOutputs().upToDateWhen(o -> false);
+        });
+
     }
 }