diff --git a/bazel/tests/testdata/BUILD b/bazel/tests/testdata/BUILD
index a49aa84393..47a3a90ac5 100644
--- a/bazel/tests/testdata/BUILD
+++ b/bazel/tests/testdata/BUILD
@@ -1,3 +1,4 @@
+load("//bazel:proto_library.bzl", "proto_library")
 load("//bazel/toolchains:proto_lang_toolchain.bzl", "proto_lang_toolchain")
 
 package(
diff --git a/editions/codegen_tests/BUILD b/editions/codegen_tests/BUILD
index 1ee0a9ef50..589ee5583d 100644
--- a/editions/codegen_tests/BUILD
+++ b/editions/codegen_tests/BUILD
@@ -1,4 +1,5 @@
 load("//bazel:cc_proto_library.bzl", "cc_proto_library")
+load("//bazel:proto_library.bzl", "proto_library")
 
 package(
     default_applicable_licenses = ["//:license"],
diff --git a/rust/test/BUILD b/rust/test/BUILD
index a1049d75b3..6bbe870e37 100644
--- a/rust/test/BUILD
+++ b/rust/test/BUILD
@@ -4,6 +4,7 @@
 # license that can be found in the LICENSE file or at
 # https://developers.google.com/open-source/licenses/bsd
 
+load("//bazel:proto_library.bzl", "proto_library")
 load(
     "//rust:defs.bzl",
     "rust_cc_proto_library",
diff --git a/rust/test/cpp/BUILD b/rust/test/cpp/BUILD
index 04cef51f3d..3293654847 100644
--- a/rust/test/cpp/BUILD
+++ b/rust/test/cpp/BUILD
@@ -12,6 +12,7 @@
 #   `//rust:protobuf`.
 
 load("@rules_rust//rust:defs.bzl", "rust_test")
+load("//bazel:proto_library.bzl", "proto_library")
 load(
     "//rust:defs.bzl",
     "rust_cc_proto_library",
diff --git a/rust/test/shared/utf8/BUILD b/rust/test/shared/utf8/BUILD
index 51f3b4da01..6f5ee3b2b7 100644
--- a/rust/test/shared/utf8/BUILD
+++ b/rust/test/shared/utf8/BUILD
@@ -1,5 +1,6 @@
 load("@rules_rust//rust:defs.bzl", "rust_test")
 load("//bazel:cc_proto_library.bzl", "cc_proto_library")
+load("//bazel:proto_library.bzl", "proto_library")
 load("//rust:defs.bzl", "rust_cc_proto_library", "rust_upb_proto_library")
 
 licenses(["notice"])