diff --git a/docs/mindformers/docs/source_en/feature/dataset.md b/docs/mindformers/docs/source_en/feature/dataset.md
index 0775ac430eac610be0d0bcfa8ea7ba960f13b4fc..5cde1a35cad3aba3023c874e8142d6977872bafc 100644
--- a/docs/mindformers/docs/source_en/feature/dataset.md
+++ b/docs/mindformers/docs/source_en/feature/dataset.md
@@ -408,7 +408,7 @@ If using a `Class`:
 
    ```yaml
    handler:
-     - type: 'custom_process'
+     - type: CustomHandler
        seed: 42
    ```
 
@@ -432,7 +432,7 @@ If using a `Method`:
 
    ```yaml
    handler:
-     custom_process:
+     - type: custom_process
        seed: 42
    ```
 
diff --git a/docs/mindformers/docs/source_zh_cn/feature/dataset.md b/docs/mindformers/docs/source_zh_cn/feature/dataset.md
index 72f42c590c425ddffc89465be2c0c61f10326f42..3cc754841544cccec1560eec99669fabb261a950 100644
--- a/docs/mindformers/docs/source_zh_cn/feature/dataset.md
+++ b/docs/mindformers/docs/source_zh_cn/feature/dataset.md
@@ -397,7 +397,7 @@ handler:
 
    ```yaml
    handler:
-     CustomHandler:
+     - type: CustomHandler
        seed: 42
    ```
 
@@ -421,7 +421,7 @@ handler:
 
    ```yaml
    handler:
-     - type: 'custom_process'
+     - type: custom_process
        seed: 42
    ```