diff --git a/ballerina-shell/modules/shell-cli/build.gradle b/ballerina-shell/modules/shell-cli/build.gradle
index a10e54d4dbec..561ccb4ee138 100644
--- a/ballerina-shell/modules/shell-cli/build.gradle
+++ b/ballerina-shell/modules/shell-cli/build.gradle
@@ -52,7 +52,7 @@ test {
 
 def mainCliClass = 'io.ballerina.shell.cli.ReplShellApplication'
 task run(type: JavaExec) {
-    main = mainCliClass
+    mainClass = mainCliClass
     standardInput = System.in
     classpath = sourceSets.main.runtimeClasspath
 }
diff --git a/compiler/ballerina-treegen/build.gradle b/compiler/ballerina-treegen/build.gradle
index c0ec997c7c88..4d969290998c 100644
--- a/compiler/ballerina-treegen/build.gradle
+++ b/compiler/ballerina-treegen/build.gradle
@@ -21,7 +21,7 @@ apply from: "$rootDir/gradle/javaProject.gradle"
 description = 'Ballerina - Syntax Tree Generator'
 
 task treegen(type: JavaExec) {
-    main = "io.ballerinalang.compiler.internal.treegen.TreeGen"
+    mainClass = "io.ballerinalang.compiler.internal.treegen.TreeGen"
     classpath = sourceSets.main.runtimeClasspath
 }
 
diff --git a/distribution/zip/jballerina-tools/build.gradle b/distribution/zip/jballerina-tools/build.gradle
index e9cef27a1bcf..fb959ff2325a 100644
--- a/distribution/zip/jballerina-tools/build.gradle
+++ b/distribution/zip/jballerina-tools/build.gradle
@@ -413,7 +413,7 @@ task generateDocs(type: JavaExec) {
     outputs.cacheIf { true }
     systemProperty("ballerina.home", "$buildDir")
     classpath = files(pathingJar.archivePath)
-    main = 'org.ballerinalang.plugin.gradle.doc.DocerinaGen'
+    mainClass = 'org.ballerinalang.plugin.gradle.doc.DocerinaGen'
     args("$buildDir/api-docs")
 }
 
diff --git a/docs/bir-spec/build.gradle b/docs/bir-spec/build.gradle
index f90d1439b2fa..cbcf1d92f316 100644
--- a/docs/bir-spec/build.gradle
+++ b/docs/bir-spec/build.gradle
@@ -59,7 +59,7 @@ checkstyleMain.exclude '**/kaitai/**'
 task genBirSpec(type: JavaExec) {
     dependsOn classes
     classpath += sourceSets.main.runtimeClasspath
-    main = 'org.ballerinalang.birspec.BIRSpecGenerator'
+    mainClass = 'org.ballerinalang.birspec.BIRSpecGenerator'
 
     inputs.dir('src/main/resources')
     outputs.files('../compiler/bir-spec.md')
diff --git a/gradle/balNativeLibProject.gradle b/gradle/balNativeLibProject.gradle
index 911c06ef3f66..8d35c5936c4d 100644
--- a/gradle/balNativeLibProject.gradle
+++ b/gradle/balNativeLibProject.gradle
@@ -50,7 +50,7 @@ class CreateBalaTask extends JavaExec {
     def newParser = 'true'
 
     CreateBalaTask() {
-        setMain('org.ballerinalang.stdlib.utils.GenerateBala')
+        setMainClass('org.ballerinalang.stdlib.utils.GenerateBala')
         setEnableAssertions(true)
     }
 
diff --git a/gradle/ballerinaNativeStdLibBuild.gradle b/gradle/ballerinaNativeStdLibBuild.gradle
index 4d63b66c8239..4e3e7cdcc914 100644
--- a/gradle/ballerinaNativeStdLibBuild.gradle
+++ b/gradle/ballerinaNativeStdLibBuild.gradle
@@ -83,7 +83,7 @@ class BallerinaLangLibBuildTask extends JavaExec {
     @Internal def skipBootstrap = 'false'
 
     BallerinaLangLibBuildTask() {
-        setMain('org.ballerinalang.stdlib.utils.BuildLangLib')
+        setMainClass('org.ballerinalang.stdlib.utils.BuildLangLib')
     }
 
     @Override
diff --git a/gradle/ballerinaStdLibBuildInternal.gradle b/gradle/ballerinaStdLibBuildInternal.gradle
index ae778da45d58..2415c2185ad6 100644
--- a/gradle/ballerinaStdLibBuildInternal.gradle
+++ b/gradle/ballerinaStdLibBuildInternal.gradle
@@ -85,7 +85,7 @@ class BallerinaLangLibBuildTask extends JavaExec {
     @Internal def skipBootstrap = 'false'
 
     BallerinaLangLibBuildTask() {
-        setMain('org.ballerinalang.stdlib.utils.BuildLangLib')
+        setMainClass('org.ballerinalang.stdlib.utils.BuildLangLib')
     }
 
     @Override