diff --git a/tools/bzl/jmh.bzl b/tools/bzl/jmh.bzl
index eee9ed0..a51a42c 100644
--- a/tools/bzl/jmh.bzl
+++ b/tools/bzl/jmh.bzl
@@ -41,6 +41,8 @@
 
 # Definitions to run jmh microbenchmarks
 
+load("@rules_java//java:defs.bzl", "java_binary", "java_plugin")
+
 def jmh_java_benchmarks(name, srcs, deps = [], tags = [], plugins = [], **kwargs):
     """Builds runnable JMH benchmarks.
     This rule builds a runnable target for one or more JMH benchmarks
@@ -48,14 +50,14 @@
     except for main_class.
     """
     plugin_name = "_{}_jmh_annotation_processor".format(name)
-    native.java_plugin(
+    java_plugin(
         name = plugin_name,
         deps = ["//lib/jmh:jmh"],
         processor_class = "org.openjdk.jmh.generators.BenchmarkProcessor",
         visibility = ["//visibility:private"],
         tags = tags,
     )
-    native.java_binary(
+    java_binary(
         name = name,
         srcs = srcs,
         main_class = "org.openjdk.jmh.Main",
@@ -63,4 +65,4 @@
         plugins = plugins + [plugin_name],
         tags = tags,
         **kwargs
-    )
\ No newline at end of file
+    )