diff --git a/src/test/java/com/google/devtools/build/lib/packages/util/MockProtoSupport.java b/src/test/java/com/google/devtools/build/lib/packages/util/MockProtoSupport.java
index c4daec8..76b55f9 100644
--- a/src/test/java/com/google/devtools/build/lib/packages/util/MockProtoSupport.java
+++ b/src/test/java/com/google/devtools/build/lib/packages/util/MockProtoSupport.java
@@ -138,12 +138,12 @@
     config.create("net/grpc/BUILD",
         "package(default_visibility=['//visibility:public'])",
         "cc_library(name = 'grpc++_codegen_lib')");
-    config.create("net/rpc/python/BUILD",
-        "package(default_visibility=['//visibility:public'])",
-        "py_library(name = 'proto_python_api_1_stub',",
-        "           srcs = [ 'test_only_prefix_proto_python_api_1_stub.py' ])",
+    config.create(
+        "net/rpc/python/BUILD",
         "py_library(name = 'proto_python_api_2_stub',",
-        "           srcs = [ 'test_only_prefix_proto_python_api_2_stub.py' ])");
+        "           srcs = [ 'test_only_prefix_proto_python_api_2_stub.py' ],",
+        "           visibility = ['//visibility:public'],",
+        ")");
     config.create("java/com/google/net/rpc/BUILD",
         "package(default_visibility=['//visibility:public'])",
         "java_library(name = 'rpc',",