diff --git a/build.gradle b/build.gradle
index b243ec7c927192b099cf7ea0d0782ba519c9f0a6..c925fcb7d848343115df4e2352459eb6e513259d 100644
--- a/build.gradle
+++ b/build.gradle
@@ -99,6 +99,7 @@ def grammarDiagramFile = './src/gen/resources/diagrams/grammar/openapiRelast.png
 task generateGrammarDiagrams(type: JavaExec) {
     group = 'Documentation'
     classpath = configurations.grammar2uml
+    main = 'de.tudresden.inf.st.jastadd.grammar2uml.compiler.Compiler'
 
     args "--output=${grammarDiagramFile}", '--defaultFolders'
     args genAst
@@ -183,3 +184,6 @@ jastadd {
         delete "src/gen-res"
     }
 }
+
+generateAst.dependsOn relast
+generateGrammarDiagrams.dependsOn generateAst
diff --git a/gradle.properties b/gradle.properties
index c0b50526cb7547533c933897a1db274967747899..99c154b69102a4a8b2843c255511324e99e988d8 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -3,4 +3,4 @@ swagger_parser_version = 2.0.30
 junit_jupiter_version = 5.7.0
 json_path_version = 2.6.0
 jastaddgradle_version = 1.13.3
-grammar2uml_version = 0.2.1
\ No newline at end of file
+grammar2uml_version = 0.2.2-13