diff --git a/struct2tensor/benchmarks/BUILD b/struct2tensor/benchmarks/BUILD
index ef15493..a12cec2 100644
--- a/struct2tensor/benchmarks/BUILD
+++ b/struct2tensor/benchmarks/BUILD
@@ -16,6 +16,7 @@
 # Placeholder: load py_binary
 # Placeholder: load py_library
 # Placeholder: load py_test
+load("//third_party/protobuf/bazel:proto_library.bzl", "proto_library")
 load("//tools/build_defs/proto/cpp:cc_proto_library.bzl", "cc_proto_library")
 
 licenses(["notice"])