diff --git a/application/build.gradle.kts b/application/build.gradle.kts
index ff1b025..ff9666e 100644
--- a/application/build.gradle.kts
+++ b/application/build.gradle.kts
@@ -1,12 +1,14 @@
 plugins {
-    application
+    id("application")
     id("com.github.johnrengelman.shadow")
 }
 
 application {
-    mainClassName = "com.harishkannarao.kotlin.exercise.DemoApplication"
+    mainClass.set("com.harishkannarao.kotlin.exercise.DemoApplication")
 }
 
+project.setProperty("mainClassName", application.mainClass.get())
+
 tasks {
     named<JavaExec>("run") {
         val properties = System.getProperties().entries.associate { it.key.toString() to it.value }