diff --git a/src/main/kotlin/xyz/wagyourtail/patchbase/gradle/PatchCreatorPlugin.kt b/src/main/kotlin/xyz/wagyourtail/patchbase/gradle/PatchCreatorPlugin.kt
index 0c9ba91..de92eb4 100644
--- a/src/main/kotlin/xyz/wagyourtail/patchbase/gradle/PatchCreatorPlugin.kt
+++ b/src/main/kotlin/xyz/wagyourtail/patchbase/gradle/PatchCreatorPlugin.kt
@@ -62,7 +62,7 @@ fun MinecraftConfig.patchBaseCreator() {
         }
     }
 
-    project.tasks.register("createClassPatch", CreateClassPatchTask::class.java) {
+    project.tasks.register("createClassPatch".withSourceSet(sourceSet), CreateClassPatchTask::class.java) {
         it.group = "patchbase"
         it.inputFile.set((project.tasks.findByName("remapJar".withSourceSet(sourceSet)) as Jar).outputs.files.singleFile)