diff --git a/docs/source/extensions/datamodules.rst b/docs/source/extensions/datamodules.rst
index b1b780c3e1..b9860be797 100644
--- a/docs/source/extensions/datamodules.rst
+++ b/docs/source/extensions/datamodules.rst
@@ -303,7 +303,7 @@ Returns a special dataloader for inference. This is the dataloader that the Trai
 transfer_batch_to_device
 ^^^^^^^^^^^^^^^^^^^^^^^^
 Override to define how you want to move an arbitrary batch to a device.
-To check the current state of execution of this hook you can use ``self.trainer.training/testing/validating/predicting``
+To check the current state of execution of this hook you can use ``self.trainer.training/testing/validating/predicting/sanity_checking``
 so that you can add different logic as per your requirement.
 
 .. testcode::
@@ -322,7 +322,7 @@ so that you can add different logic as per your requirement.
 on_before_batch_transfer
 ^^^^^^^^^^^^^^^^^^^^^^^^
 Override to alter or apply augmentations to your batch before it is transferred to the device.
-To check the current state of execution of this hook you can use ``self.trainer.training/testing/validating/predicting``
+To check the current state of execution of this hook you can use ``self.trainer.training/testing/validating/predicting/sanity_checking``
 so that you can add different logic as per your requirement.
 
 .. testcode::
@@ -339,7 +339,7 @@ so that you can add different logic as per your requirement.
 on_after_batch_transfer
 ^^^^^^^^^^^^^^^^^^^^^^^
 Override to alter or apply augmentations to your batch after it is transferred to the device.
-To check the current state of execution of this hook you can use ``self.trainer.training/testing/validating/predicting``
+To check the current state of execution of this hook you can use ``self.trainer.training/testing/validating/predicting/sanity_checking``
 so that you can add different logic as per your requirement.
 
 .. testcode::