diff --git a/docs/BUILD b/docs/BUILD
index 8e85dbf86..80d3ab550 100644
--- a/docs/BUILD
+++ b/docs/BUILD
@@ -4,12 +4,10 @@ py_binary(
     name = "build_py_api_docs",
     srcs = ["build_py_api_docs.py"],
     deps = [
-        "//mediapipe",
         "//third_party/py/absl:app",
         "//third_party/py/absl/flags",
-        "//third_party/py/tensorflow_docs",
+        "//third_party/py/mediapipe",
         "//third_party/py/tensorflow_docs/api_generator:generate_lib",
-        "//third_party/py/tensorflow_docs/api_generator:public_api",
     ],
 )