diff --git a/doc/source/raydp.rst b/doc/source/raydp.rst
index cee142344..9a8353ccc 100644
--- a/doc/source/raydp.rst
+++ b/doc/source/raydp.rst
@@ -75,9 +75,9 @@ Training a Spark DataFrame with TensorFlow
   tensorflow_model = estimator.get_model()
 
 
-^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
-Training a Spark DataFrame with TensorFlow
-^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+Training a Spark DataFrame with PyTorch
+^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 
 Similarly, ``raydp.torch.TorchEstimator`` provides an API for training with
 PyTorch.