diff --git a/example/build.gradle b/example/build.gradle
index f7f95e5..0719011 100644
--- a/example/build.gradle
+++ b/example/build.gradle
@@ -2,7 +2,7 @@ apply plugin: 'java'
 apply plugin: 'application'
 
 application {
-    mainClassName = 'example.OpenAiApiExample'
+    mainClass.set('example.OpenAiApiExample')
 }
 
 dependencies {