diff --git a/build.gradle b/build.gradle
index e930214..9898782 100644
--- a/build.gradle
+++ b/build.gradle
@@ -1,5 +1,6 @@
 plugins {
     id "java"
+    id "application"
     id "maven-publish"
     id "net.kyori.blossom" version "1.3.1"
 }
@@ -129,6 +130,10 @@ java {
     withSourcesJar()
 }
 
+application {
+    mainClass = "net.raphimc.viaproxy.ViaProxy"
+}
+
 jar {
     dependsOn configurations.include
     from {
@@ -142,7 +147,7 @@ jar {
 
     manifest {
         attributes(
-                "Main-Class": "net.raphimc.viaproxy.ViaProxy",
+                "Main-Class": application.mainClass,
                 "Multi-Release": "true",
                 "Launcher-Agent-Class": "net.raphimc.viaproxy.ViaProxy"
         )