diff --git a/test/torchaudio_unittest/prototype/transforms/batch_consistency_test.py b/test/torchaudio_unittest/prototype/transforms/batch_consistency_test.py
index 9674f9a77d..2bb47f55a3 100644
--- a/test/torchaudio_unittest/prototype/transforms/batch_consistency_test.py
+++ b/test/torchaudio_unittest/prototype/transforms/batch_consistency_test.py
@@ -2,7 +2,6 @@
 
 import torch
 import torchaudio.prototype.transforms as T
-import torchaudio.transforms as transforms
 from torchaudio_unittest.common_utils import TorchaudioTestCase
 
 
diff --git a/torchaudio/_internal/module_utils.py b/torchaudio/_internal/module_utils.py
index b620122c70..d5ab186b0c 100644
--- a/torchaudio/_internal/module_utils.py
+++ b/torchaudio/_internal/module_utils.py
@@ -46,6 +46,7 @@ def deprecated(direction: str, version: Optional[str] = None, remove: bool = Fal
     Args:
         direction (str): Migration steps to be given to users.
         version (str or int): The version when the object will be removed
+        remove (bool): If enabled, append future removal message.
     """
 
     def decorator(func):