diff --git a/tools/test_r8cfsegments.py b/tools/test_r8cfsegments.py
index 20a0448..426718a 100755
--- a/tools/test_r8cfsegments.py
+++ b/tools/test_r8cfsegments.py
@@ -58,11 +58,11 @@
 
 
 def Main():
-  utils.check_java_version()
   args = parse_arguments()
-  output_dir = args.output
   if args.golem:
     golem.link_third_party()
+  utils.check_java_version()
+  output_dir = args.output
   with utils.TempDir() as temp_dir:
     if not output_dir:
       output_dir = temp_dir