diff --git a/build.gradle.kts b/build.gradle.kts
index 24f205035..4cd8f41fe 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -1,4 +1,4 @@
-import xyz.jpenilla.toothpick.task.ApplyPatches.Companion.DEFAULT_APPLY_COMMAND
+import xyz.jpenilla.toothpick.task.ApplyPatches
 
 plugins {
     `java-library`
@@ -48,7 +48,7 @@ subprojects {
 }
 
 tasks {
-    applyPatches {
-        applyCommand(DEFAULT_APPLY_COMMAND.toMutableList().apply { add("--keep-non-patch") })
+    withType<ApplyPatches> {
+       argument("--keep-non-patch")
     }
 }