diff --git a/python/internal.bzl b/python/internal.bzl
index 8f99becec..8c667e1ae 100644
--- a/python/internal.bzl
+++ b/python/internal.bzl
@@ -1,6 +1,7 @@
 """
 Internal helpers for building the Python protobuf runtime.
 """
+load("@rules_python//python:py_test.bzl", "py_test")
 
 def _remove_cross_repo_path(path):
     components = path.split("/")
@@ -123,7 +124,7 @@ def internal_py_test(deps = [], **kwargs):
       deps: any additional dependencies of the test.
       **kwargs: arguments forwarded to py_test.
     """
-    native.py_test(
+    py_test(
         imports = ["."],
         deps = deps + ["//python:python_test_lib"],
         target_compatible_with = select({
diff --git a/python/pb_unit_tests/pyproto_test_wrapper.bzl b/python/pb_unit_tests/pyproto_test_wrapper.bzl
index b5f474d3d..1e2b5daa2 100644
--- a/python/pb_unit_tests/pyproto_test_wrapper.bzl
+++ b/python/pb_unit_tests/pyproto_test_wrapper.bzl
@@ -1,8 +1,10 @@
 """Wrapper for another py_test to run with upb, possibly with a set of expected failures."""
 
+load("@rules_python//python:py_test.bzl", "py_test")
+
 def pyproto_test_wrapper(name, deps = []):
     src = name + "_wrapper.py"
-    native.py_test(
+    py_test(
         name = name,
         srcs = [src],
         legacy_create_init = False,
diff --git a/python/py_extension.bzl b/python/py_extension.bzl
index 9a2828b95..2741cdf6e 100644
--- a/python/py_extension.bzl
+++ b/python/py_extension.bzl
@@ -1,6 +1,7 @@
 """Macro to support py_extension """
 
 load("@bazel_skylib//lib:selects.bzl", "selects")
+load("@rules_python//python:py_library.bzl", "py_library")
 
 def py_extension(name, srcs, copts, deps = [], **kwargs):
     """Creates a C++ library to extend python
@@ -50,7 +51,7 @@ def py_extension(name, srcs, copts, deps = [], **kwargs):
         visibility = ["//python:__subpackages__"],
     )
 
-    native.py_library(
+    py_library(
         name = name,
         data = [output_file],
         imports = ["."],
diff --git a/upb/cmake/build_defs.bzl b/upb/cmake/build_defs.bzl
index 325189c7f..c7a7a4558 100644
--- a/upb/cmake/build_defs.bzl
+++ b/upb/cmake/build_defs.bzl
@@ -7,6 +7,8 @@
 
 """Bazel support functions related to CMake support."""
 
+load("@rules_python//python:py_test.bzl", "py_test")
+
 def staleness_test(name, outs, generated_pattern, target_files = None, tags = [], **kwargs):
     """Tests that checked-in file(s) match the contents of generated file(s).
 
@@ -46,7 +48,7 @@ def staleness_test(name, outs, generated_pattern, target_files = None, tags = []
               "sed -i.bak -e 's|INSERT_FILE_LIST_HERE|" + "\\\n  ".join(file_list) + "|' $@",
     )
 
-    native.py_test(
+    py_test(
         name = name,
         srcs = [script_name],
         data = existing_outs + [generated_pattern % file for file in outs],