diff --git a/covid19-model/build.gradle b/covid19-model/build.gradle
index 5271077..b65fdc2 100644
--- a/covid19-model/build.gradle
+++ b/covid19-model/build.gradle
@@ -13,5 +13,9 @@ dependencies {
     testImplementation 'com.h2database:h2:1.4.194'
 }
 
+springBoot {
+    mainClassName 'com.bvn13.covid19.model.Covid19ModelApplication'
+}
+
 bootJar { enabled = true }
 jar { enabled = true }
\ No newline at end of file