diff --git a/build.gradle b/build.gradle
index 3e167e3d..fe9c1335 100644
--- a/build.gradle
+++ b/build.gradle
@@ -67,6 +67,17 @@ loom {
     }
 }
 
+sourceSets {
+    main {
+        // Add the datagenned files into the jar.
+        resources {
+            srcDirs += [
+                    'src/main/generated'
+            ]
+        }
+    }
+}
+
 dependencies {
     minecraft "com.mojang:minecraft:${project.minecraft_version}"
     mappings loom.officialMojangMappings()