diff --git a/groovy/groovy.bzl b/groovy/groovy.bzl
index 91d9e77..cb67595 100644
--- a/groovy/groovy.bzl
+++ b/groovy/groovy.bzl
@@ -225,7 +225,8 @@ def _groovy_test_impl(ctx):
     classes = [path_to_class(src.path) for src in ctx.files.srcs]
 
     # Write a file that executes JUnit on the inferred classes
-    cmd = "$JAVA_HOME/bin/java %s -cp %s org.junit.runner.JUnitCore %s\n" % (
+    cmd = "%s/bin/java %s -cp %s org.junit.runner.JUnitCore %s\n" % (
+        ctx.attr._jdk[java_common.JavaRuntimeInfo].java_home,
         " ".join(ctx.attr.jvm_flags),
         ":".join([dep.short_path for dep in all_deps.to_list()]),
         " ".join(classes),