diff --git a/bazel/pyproto_test_wrapper.bzl b/bazel/pyproto_test_wrapper.bzl
index a8b46156b0..fb909a1db8 100644
--- a/bazel/pyproto_test_wrapper.bzl
+++ b/bazel/pyproto_test_wrapper.bzl
@@ -18,14 +18,17 @@ def pyproto_test_wrapper(name):
     )
 
 # copybara:replace_for_google3_begin
-# 
+#
 # def pyproto_test_wrapper(name):
 #     src = name + "_wrapper.py"
 #     native.py_test(
 #         name = name,
 #         srcs = [src],
 #         main = src,
-#         deps = ["//net/proto2/python/internal:" + name + "_for_deps"],
+#         deps = [
+#             "//net/proto2/python/internal:" + name + "_for_deps",
+#             "//net/proto2/python/public:use_upb_protos",
+#         ],
 #     )
-# 
+#
 # copybara:replace_for_google3_end