diff --git a/demo/build.gradle b/demo/build.gradle
index 92d3f51..7d88c38 100644
--- a/demo/build.gradle
+++ b/demo/build.gradle
@@ -1,6 +1,6 @@
-plugins {
-    id 'java-library'
-}
+apply plugin: 'application'
+
+mainClassName = 'App'
 
 sourceCompatibility = JavaVersion.VERSION_1_8
 targetCompatibility = JavaVersion.VERSION_1_8