diff --git a/build.gradle b/build.gradle
index a8819368b..92f2e7a4c 100755
--- a/build.gradle
+++ b/build.gradle
@@ -67,6 +67,12 @@ loom {
         }
     }
     mixin.defaultRefmapName = "mixins.baritone.refmap.json"
+
+    runs {
+        client {
+            source = sourceSets.launch
+        }
+    }
 }