diff --git a/build.gradle b/build.gradle
index 285777fc..69922b16 100644
--- a/build.gradle
+++ b/build.gradle
@@ -28,18 +28,13 @@ archivesBaseName = project.name
 loom {
     mixin.defaultRefmapName = 'unicopia.mixin.refmap.json'
     accessWidenerPath = file('src/main/resources/unicopia.aw')
-    runs {
-        datagen {
-            server()
-            name "Data Generation"
-            vmArg "-Dfabric-api.datagen"
-            vmArg "-Dfabric-api.datagen.modid=unicopia"
-            vmArg "-Dfabric-api.datagen.output-dir=${file("src/main/generated")}"
-            runDir "build/datagen"
-        }
+}
+
+fabricApi {
+    configureDataGeneration {
+      modId = 'unicopia'
     }
 }
-//assemble.dependsOn(runDatagen)
 
 reckon {
     scopeFromProp()
@@ -108,14 +103,6 @@ dependencies {
     }
 }
 
-sourceSets {
-  main {
-    resources {
-      srcDirs += [ "src/main/generated" ]
-    }
-  }
-}
-
 processResources {
     inputs.property "version", project.version.toString()