diff --git a/docs/source-pytorch/upgrade/sections/1_9_advanced.rst b/docs/source-pytorch/upgrade/sections/1_9_advanced.rst
index 21611ee312af0..782e16348d3b1 100644
--- a/docs/source-pytorch/upgrade/sections/1_9_advanced.rst
+++ b/docs/source-pytorch/upgrade/sections/1_9_advanced.rst
@@ -231,15 +231,15 @@
      - `PR16750`_
 
    * - had any logic except reducing the DP outputs in  ``LightningModule.training_step_end`` hook
-     - port it to ``LightningModule.training_batch_end`` hook
+     - port it to ``LightningModule.on_train_batch_end`` hook
      - `PR16791`_
 
    * - had any logic except reducing the DP outputs in  ``LightningModule.validation_step_end`` hook
-     - port it to ``LightningModule.validation_batch_end`` hook
+     - port it to ``LightningModule.on_validation_batch_end`` hook
      - `PR16791`_
 
    * - had any logic except reducing the DP outputs in  ``LightningModule.test_step_end`` hook
-     - port it to ``LightningModule.test_batch_end`` hook
+     - port it to ``LightningModule.on_test_batch_end`` hook
      - `PR16791`_
 
    * - used ``pl.strategies.DDPSpawnStrategy``