diff --git a/tensorflow_text/BUILD b/tensorflow_text/BUILD
index 8078fd391..93b663040 100644
--- a/tensorflow_text/BUILD
+++ b/tensorflow_text/BUILD
@@ -77,6 +77,7 @@ py_library(
         "python/ops/__init__.py",
         "tools/__init__.py",
     ],
+    lazy_imports = True,
     deps = [
         ":bert_tokenizer",
         ":boise_offset_converter",