diff --git a/pgpainless-cli/build.gradle b/pgpainless-cli/build.gradle
index 869eedf7..7317524b 100644
--- a/pgpainless-cli/build.gradle
+++ b/pgpainless-cli/build.gradle
@@ -48,6 +48,10 @@ dependencies {
 
 mainClassName = 'org.pgpainless.cli.PGPainlessCLI'
 
+application {
+    mainClass = mainClassName
+}
+
 jar {
     duplicatesStrategy(DuplicatesStrategy.EXCLUDE)
     manifest {