diff --git a/xla/hlo/testlib/BUILD b/xla/hlo/testlib/BUILD
index 46fb09b05f1c6..a1f34221afe87 100644
--- a/xla/hlo/testlib/BUILD
+++ b/xla/hlo/testlib/BUILD
@@ -66,6 +66,7 @@ cc_library(
         "//xla:shape_util",
         "//xla:util",
         "//xla:xla_data_proto_cc",
+        "//xla:xla_proto_cc",
         "//xla/hlo/ir:hlo",
         "//xla/hlo/ir:hlo_module_group",
         "//xla/hlo/pass:hlo_pass",
diff --git a/xla/hlo/testlib/hlo_hardware_independent_test_base.cc b/xla/hlo/testlib/hlo_hardware_independent_test_base.cc
index 9fef4f365e390..b49b4356d055b 100644
--- a/xla/hlo/testlib/hlo_hardware_independent_test_base.cc
+++ b/xla/hlo/testlib/hlo_hardware_independent_test_base.cc
@@ -51,6 +51,8 @@ limitations under the License.
 #include "xla/tsl/platform/statusor.h"
 #include "xla/tsl/platform/test.h"
 #include "xla/util.h"
+#include "xla/xla.pb.h"
+#include "xla/xla_data.pb.h"
 
 namespace xla {
 
diff --git a/xla/hlo/testlib/hlo_hardware_independent_test_base.h b/xla/hlo/testlib/hlo_hardware_independent_test_base.h
index 90ddb142f0c29..9fb276d16c790 100644
--- a/xla/hlo/testlib/hlo_hardware_independent_test_base.h
+++ b/xla/hlo/testlib/hlo_hardware_independent_test_base.h
@@ -45,6 +45,7 @@ limitations under the License.
 #include "xla/shape_util.h"
 #include "xla/tsl/platform/test.h"
 #include "xla/util.h"
+#include "xla/xla.pb.h"
 #include "xla/xla_data.pb.h"
 
 namespace xla {