diff --git a/build.gradle b/build.gradle
index c05d6a148..7d21a7211 100644
--- a/build.gradle
+++ b/build.gradle
@@ -235,7 +235,7 @@ tasks.register('ktlint', JavaExec) {
     group = "verification"
     description = "Check Kotlin code style."
     classpath = configurations.ktlint
-    main = "com.pinterest.ktlint.Main"
+    mainClass.set("com.pinterest.ktlint.Main")
     args "app/src/**/*.kt", "!**/build/**/*.kt", "--baseline=ktlint-baseline.xml"
 }