diff --git a/scripts/checkcliargs.sh b/scripts/checkcliargs.sh
index 7394fc1..f426f6b 100755
--- a/scripts/checkcliargs.sh
+++ b/scripts/checkcliargs.sh
@@ -1,17 +1,17 @@
 #!/bin/bash
 
-TMPNEW=/tmp/checkcliargs-new
-TMPOLD=/tmp/checkcliargs-old
+main=org.jetbrains.dokka.MainKt
+new=/tmp/checkcliargs-new
+old=/tmp/checkcliargs-old
 
-java -cp "lib/compile/*" -jar lib/compile/dokka-cli-*.jar -h >$TMPNEW
-java -cp "/examples/lib/bld*" -jar examples/lib/bld/dokka-cli-*.jar -h >$TMPOLD
+java -cp "lib/compile/*" $main -h >$new
+java -cp "/examples/lib/bld*" $main -h >$old
 
-diff $TMPOLD $TMPNEW
+diff $old $new
 
-java -cp "lib/compile/*" -jar lib/compile/dokka-cli-*.jar -sourceSet -h >$TMPNEW
-java -cp "/examples/lib/bld*" -jar examples/lib/bld/dokka-cli-*.jar -sourceSet -h >$TMPOLD
+java -cp "lib/compile/*" $main -sourceSet -h >$new
+java -cp "/examples/lib/bld*" $main -sourceSet -h >$old
 
-diff $TMPOLD $TMPNEW
+diff $old $new
 
-
-rm -rf $TMPNEW $TMPOLD
+rm -rf $new $old
diff --git a/scripts/cliargs.sh b/scripts/cliargs.sh
index dda5b47..0fd649a 100755
--- a/scripts/cliargs.sh
+++ b/scripts/cliargs.sh
@@ -1,11 +1,13 @@
 #!/bin/bash
 
-java -cp "lib/compile/*" -jar lib/compile/dokka-cli-*.jar -h |\
+main=org.jetbrains.dokka.MainKt
+
+java -cp "lib/compile/*" $main -h |\
 grep "    -" |\
 sed -e "s/^    -/-/" -e "s/ \[.*//" -e "s/ ->.*//" -e '/help/d' |\
 sort > "src/test/resources/dokka-args.txt"
 
-java -cp "lib/compile/*" -jar lib/compile/dokka-cli-*.jar -sourceSet -h |\
+java -cp "lib/compile/*" $main -sourceSet -h |\
 grep "    -" |\
 sed -e "s/^    -/-/" -e "s/ \[.*//" -e "s/ ->.*//" -e '/help/d' -e '/includeNonPublic/d' |\
 sort > "src/test/resources/dokka-sourceset-args.txt"