1 files changed, 7 insertions, 2 deletions

diff --git a/enigma-cli/build.gradle b/enigma-cli/build.gradle
index 8de307e..3490613 100644
--- a/enigma-cli/build.gradle
+++ b/enigma-cli/build.gradle
@@ -1,4 +1,5 @@
 plugins {
+    id 'application'
     id 'com.github.johnrengelman.shadow' version '5.2.0'
 }
 
@@ -6,7 +7,11 @@ dependencies {
     implementation project(':enigma')
 }
 
-jar.manifest.attributes 'Main-Class': 'cuchaz.enigma.command.Main'
+application {
+    mainClass = 'cuchaz.enigma.command.Main'
+}
+
+jar.manifest.attributes 'Main-Class': application.mainClass
 
 publishing {
     publications {
@@ -15,4 +20,4 @@ publishing {
             publication.artifact shadowJar
         }
     }
-}
\ No newline at end of file
+}