diff --git a/forge/build.gradle b/forge/build.gradle
index ea90264..374a5ba 100644
--- a/forge/build.gradle
+++ b/forge/build.gradle
@@ -25,6 +25,7 @@ legacyForge {
         }
         data {
             data()
+            programArguments.addAll '--mod', project.mod_id, '--all', '--output', file('src/generated/resources/').getAbsolutePath(), '--existing', file('src/main/resources/').getAbsolutePath()
         }
         server {
             server()
@@ -32,7 +33,7 @@ legacyForge {
     }
 
     mods {
-        testproject {
+        "${mod_id}" {
             sourceSet sourceSets.main
         }
     }