diff --git a/tensorflow_estimator/BUILD b/tensorflow_estimator/BUILD
index 5060a179..66ff0fcc 100644
--- a/tensorflow_estimator/BUILD
+++ b/tensorflow_estimator/BUILD
@@ -19,6 +19,8 @@ package_group(
         "//learning/brain/...",
         "//learning/deepmind/research/...",
         "//learning/tfx/models/uplift/estimators/...",
+        "//nlp/nlx/ads/expmatch/model/...",
+        "//nlp/nlx/common/query_bert/...",
         "//nlp/nlx/i18n/pangloss/...",
         "//tensorflow_estimator/...",
         "//third_party/py/tensorflow_privacy/...",