diff --git a/bazel/build_rules/proto_descriptor.bzl b/bazel/build_rules/proto_descriptor.bzl
index f114e26eee4..bf08f71b38c 100644
--- a/bazel/build_rules/proto_descriptor.bzl
+++ b/bazel/build_rules/proto_descriptor.bzl
@@ -5,7 +5,7 @@ def paths(files):
     return [f.path for f in files.to_list()]
 
 def _impl(ctx):
-    descriptors = ctx.attr.proto_library.proto.transitive_descriptor_sets
+    descriptors = ctx.attr.proto_library[ProtoInfo].transitive_descriptor_sets
     ctx.actions.run_shell(
         inputs = descriptors,
         outputs = [ctx.outputs.out],
diff --git a/bazel/proto_crd/proto_k8s_service.bzl b/bazel/proto_crd/proto_k8s_service.bzl
index a2f8322d6f4..675082e90ee 100644
--- a/bazel/proto_crd/proto_k8s_service.bzl
+++ b/bazel/proto_crd/proto_k8s_service.bzl
@@ -6,7 +6,7 @@ load("@cloud_robotics//bazel/proto_crd:proto_crd.bzl", "proto_crd")
 load("@grpc_ecosystem_grpc_gateway//protoc-gen-swagger:defs.bzl", "protoc_gen_swagger")
 
 def _impl(ctx):
-    import_files = [f.path for f in ctx.attr.spec.proto.direct_sources]
+    import_files = [f.path for f in ctx.attr.spec[ProtoInfo].direct_sources]
     imports = ["import \"{}\";".format(f) for f in import_files]
     options = ["option {} = \"{}\";".format(k, v) for k, v in ctx.attr.file_options.items()]