diff --git a/build.gradle b/build.gradle
index b11f9db477c5902871e35bd2da6351dca2a8a3b6..952a16eb13af2dbef5a10e22991fb0570b460d95 100644
--- a/build.gradle
+++ b/build.gradle
@@ -141,6 +141,7 @@ task ragConnect(type: JavaExec) {
 task generateLoadingGrammarDiagrams(type: JavaExec) {
     group = 'Documentation'
     classpath = configurations.grammar2uml
+    mainClass = 'de.tudresden.inf.st.jastadd.grammar2uml.compiler.Compiler'
 
     args "--output=${loadingGrammarDiagramFile}", '--defaultFolders'
     args loadingRelastFiles
@@ -152,6 +153,7 @@ task generateLoadingGrammarDiagrams(type: JavaExec) {
 task generateCleaningGrammarDiagrams(type: JavaExec) {
     group = 'Documentation'
     classpath = configurations.grammar2uml
+    mainClass = 'de.tudresden.inf.st.jastadd.grammar2uml.compiler.Compiler'
 
     args "--output=${cleaningGrammarDiagramFile}", '--defaultFolders'
     args cleaningRelastFiles
diff --git a/gradle.properties b/gradle.properties
index 3888fab57551e700425fee1fd3e0fecb0fd36c70..f0bd5e8959b31a2b7e9633946c60d6ca42fe46a0 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -2,7 +2,7 @@ relast_version = 0.3.0-137
 relast2uml_version = 1.1.0
 jupyter_version = 5.8.2
 assertj_version = 3.22.0
-grammar2uml_version = 0.2.1
+grammar2uml_version = 0.2.5-21
 jastaddgradle_version = 1.14.5
 javalin_version = 4.6.3
 protobuf_version = 4.0.0-rc-2