diff --git a/tensorflow_text/python/keras/layers/__init__.py b/tensorflow_text/python/keras/layers/__init__.py
index 0f5343e16..a066c2b68 100644
--- a/tensorflow_text/python/keras/layers/__init__.py
+++ b/tensorflow_text/python/keras/layers/__init__.py
@@ -19,10 +19,14 @@
 
 # pylint: disable=wildcard-import
 from tensorflow_text.python.keras.layers.todense import *
+from tensorflow_text.python.keras.layers.tokenization_layers import *
 
 # Public symbols in the "tensorflow_text.layers" package.
 _allowed_symbols = [
     "ToDense",
+    "UnicodeScriptTokenizer",
+    "WhitespaceTokenizer",
+    "WordpieceTokenizer",
 ]
 
 remove_undocumented(__name__, _allowed_symbols)