1 files changed, 5 insertions, 3 deletions

diff --git a/enigma-cli/build.gradle b/enigma-cli/build.gradle
index 5281e9e6..05e644e1 100644
--- a/enigma-cli/build.gradle
+++ b/enigma-cli/build.gradle
@@ -1,15 +1,17 @@
 plugins {
         id 'application'
-        id 'com.github.johnrengelman.shadow' version '7.0.0'
+        id 'com.github.johnrengelman.shadow'
 }
 
 dependencies {
         implementation project(':enigma')
 }
 
-mainClassName = 'cuchaz.enigma.command.Main'
+application {
+        mainClass = 'cuchaz.enigma.command.Main'
+}
 
-jar.manifest.attributes 'Main-Class': mainClassName
+jar.manifest.attributes 'Main-Class': application.mainClass.get()
 
 publishing {
         publications {