diff --git a/official/nlp/modeling/layers/text_layers.py b/official/nlp/modeling/layers/text_layers.py
index 970b53776de157206ed2c8793978e8d584367f7d..979a7829ad0fa2d6ea736a4ad8d9ea2c228ba048 100644
--- a/official/nlp/modeling/layers/text_layers.py
+++ b/official/nlp/modeling/layers/text_layers.py
@@ -57,7 +57,7 @@ def _truncate_row_lengths(ragged_tensor: tf.RaggedTensor,
 
 
 class BertTokenizer(tf.keras.layers.Layer):
-  """Wraps BertTokenizer with pre-defined vocab as a Keras Layer.
+  """Wraps TF.Text's BertTokenizer with pre-defined vocab as a Keras Layer.
 
   Attributes:
     tokenize_with_offsets: If true, calls