1 files changed, 7 insertions, 2 deletions

diff --git a/enigma-swing/build.gradle b/enigma-swing/build.gradle
index 6d2cd5f2..f7769037 100644
--- a/enigma-swing/build.gradle
+++ b/enigma-swing/build.gradle
@@ -1,4 +1,5 @@
 plugins {
+    id 'application'
     id 'com.github.johnrengelman.shadow' version '5.2.0'
 }
 
@@ -13,7 +14,11 @@ dependencies {
     implementation 'org.drjekyll:fontchooser:2.4'
 }
 
-jar.manifest.attributes 'Main-Class': 'cuchaz.enigma.gui.Main'
+application {
+    mainClass = 'cuchaz.enigma.gui.Main'
+}
+
+jar.manifest.attributes 'Main-Class': application.mainClass
 
 publishing {
     publications {
@@ -22,4 +27,4 @@ publishing {
             publication.artifact shadowJar
         }
     }
-}
\ No newline at end of file
+}