diff --git a/docs/source/data.rst b/docs/source/data.rst
index 9754f62..fee4978 100644
--- a/docs/source/data.rst
+++ b/docs/source/data.rst
@@ -33,4 +33,12 @@ Conformer
 Classes and functions from `unimol_tools.data.conformer.py <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/data/conformer.py>`_.
 
 .. automodule:: unimol_tools.data.conformer
+   :members:
+
+Split
+-------
+
+Classes and functions from `unimol_tools.data.split.py <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/data/split.py>`_ manages the split methods in the dataset.
+
+.. automodule:: unimol_tools.data.split
    :members:
\ No newline at end of file
diff --git a/docs/source/task.rst b/docs/source/task.rst
index 472c6ea..f8eafeb 100644
--- a/docs/source/task.rst
+++ b/docs/source/task.rst
@@ -14,10 +14,3 @@ Trainer
 .. automodule:: unimol_tools.tasks.trainer
    :members:
 
-Split
--------
-
-`unimol_tools.tasks.split.py <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/tasks/split.py>`_ manages the split methods in the dataset.
-
-.. automodule:: unimol_tools.tasks.split
-   :members:
\ No newline at end of file