diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
index 32ede0ab1ad..68cd6ff2e99 100644
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@@ -418,6 +418,7 @@ py_library(
         ":cpp_shape_inference_proto_py",
         ":errors",
         ":framework_for_generated_wrappers",
+        ":function",
         ":graph_util",
         ":lib",
         ":platform",