diff --git a/internal/openapi_generator.bzl b/internal/openapi_generator.bzl
index cbc0e66..f3d65b6 100644
--- a/internal/openapi_generator.bzl
+++ b/internal/openapi_generator.bzl
@@ -124,6 +124,7 @@ def _impl(ctx):
         ),
         outputs = [declared_dir],
         tools = ctx.files._jdk,
+        toolchain = ctx.attr._jdk.label,
     )
 
     srcs = declared_dir.path