diff --git a/docs/en/advanced_guides/pipeline.md b/docs/en/advanced_guides/pipeline.md
index df98afa2..058e8139 100644
--- a/docs/en/advanced_guides/pipeline.md
+++ b/docs/en/advanced_guides/pipeline.md
@@ -134,7 +134,7 @@ train_pipeline = [
 
    ```python
    from mmcv.transforms import BaseTransform
-   from mmpretrain.datasets import TRANSFORMS
+   from mmpretrain.registry import TRANSFORMS
 
    @TRANSFORMS.register_module()
    class MyTransform(BaseTransform):
diff --git a/docs/zh_CN/advanced_guides/pipeline.md b/docs/zh_CN/advanced_guides/pipeline.md
index d6403246..99506b08 100644
--- a/docs/zh_CN/advanced_guides/pipeline.md
+++ b/docs/zh_CN/advanced_guides/pipeline.md
@@ -112,7 +112,7 @@ train_pipeline = [
 
    ```python
    from mmcv.transforms import BaseTransform
-   from mmpretrain.datasets import TRANSFORMS
+   from mmpretrain.registry import TRANSFORMS
 
    @TRANSFORMS.register_module()
    class MyTransform(BaseTransform):