diff --git a/tensorflow_privacy/privacy/estimators/dnn.py b/tensorflow_privacy/privacy/estimators/dnn.py
index cac6034..0190c57 100644
--- a/tensorflow_privacy/privacy/estimators/dnn.py
+++ b/tensorflow_privacy/privacy/estimators/dnn.py
@@ -25,7 +25,7 @@ from tensorflow_estimator.python.estimator.canned import dnn
 
 
 class DNNClassifier(tf.estimator.Estimator):
-  """DP version of DNNClassifier."""
+  """DP version of `tf.estimator.DNNClassifier`."""
 
   def __init__(
       self,
diff --git a/tensorflow_privacy/privacy/estimators/v1/dnn.py b/tensorflow_privacy/privacy/estimators/v1/dnn.py
index 8ad1b94..1f480ef 100644
--- a/tensorflow_privacy/privacy/estimators/v1/dnn.py
+++ b/tensorflow_privacy/privacy/estimators/v1/dnn.py
@@ -22,7 +22,7 @@ from tensorflow_estimator.python.estimator.canned import dnn
 
 
 class DNNClassifier(tf.estimator.Estimator):
-  """DP version of tf.estimator.DNNClassifier."""
+  """DP version of `tf.compat.v1.estimator.DNNClassifier`."""
 
   def __init__(
       self,