diff --git a/torch_xla/csrc/BUILD b/torch_xla/csrc/BUILD
index b80c54a2326..796346eb336 100644
--- a/torch_xla/csrc/BUILD
+++ b/torch_xla/csrc/BUILD
@@ -294,6 +294,7 @@ ptxla_cc_library(
         "//torch_xla/csrc/runtime:cache",
         "//torch_xla/csrc/runtime:computation_client",
         "@com_google_absl//absl/types:span",
+        "@local_config_python//:python_headers",
     ],
 )