diff --git a/docs/source/api/pgl.contrib.heter_graph.rst b/docs/source/api/pgl.contrib.heter_graph.rst
deleted file mode 100644
index e1cc7ef4adeb943f352fad96781ef572606a6b96..0000000000000000000000000000000000000000
--- a/docs/source/api/pgl.contrib.heter_graph.rst
+++ /dev/null
@@ -1,7 +0,0 @@
-pgl.contrib.heter\_graph module: Heterogenous Graph Storage
-===============================
-
-.. automodule:: pgl.contrib.heter_graph
-   :members:
-   :undoc-members:
-   :show-inheritance:
diff --git a/docs/source/api/pgl.contrib.heter_graph_wrapper.rst b/docs/source/api/pgl.contrib.heter_graph_wrapper.rst
deleted file mode 100644
index 74d9fe853df4639ee86470e67aa06c85c167cffe..0000000000000000000000000000000000000000
--- a/docs/source/api/pgl.contrib.heter_graph_wrapper.rst
+++ /dev/null
@@ -1,7 +0,0 @@
-pgl.contrib.heter\_graph\_wrapper module: Heterogenous Graph data holders for Paddle GNN.
-=========================
-
-.. automodule:: pgl.contrib.heter_graph_wrapper
-   :members:
-   :undoc-members:
-   :show-inheritance:
diff --git a/docs/source/api/pgl.heter_graph.rst b/docs/source/api/pgl.heter_graph.rst
new file mode 100644
index 0000000000000000000000000000000000000000..f3d9091f2487533d54864f684bb1683a56d70f94
--- /dev/null
+++ b/docs/source/api/pgl.heter_graph.rst
@@ -0,0 +1,7 @@
+pgl.heter\_graph module: Heterogenous Graph Storage
+===============================
+
+.. automodule:: pgl.heter_graph
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/source/api/pgl.heter_graph_wrapper.rst b/docs/source/api/pgl.heter_graph_wrapper.rst
new file mode 100644
index 0000000000000000000000000000000000000000..63e364d9d2d02f2a52f4752ab017291624cdf40d
--- /dev/null
+++ b/docs/source/api/pgl.heter_graph_wrapper.rst
@@ -0,0 +1,7 @@
+pgl.heter\_graph\_wrapper module: Heterogenous Graph data holders for Paddle GNN.
+=========================
+
+.. automodule:: pgl.heter_graph_wrapper
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/source/api/pgl.rst b/docs/source/api/pgl.rst
index cc581c85b0298d3454bcbab0779427e2b25340d9..57034dabb67a995d46993e0218aaec76a1e897fa 100644
--- a/docs/source/api/pgl.rst
+++ b/docs/source/api/pgl.rst
@@ -9,5 +9,5 @@ API Reference
    pgl.data_loader
    pgl.utils.paddle_helper
    pgl.utils.mp_reader
-   pgl.contrib.heter_graph
-   pgl.contrib.heter_graph_wrapper
+   pgl.heter_graph
+   pgl.heter_graph_wrapper
diff --git a/docs/source/quick_start/md/quick_start_for_heterGraph.md b/docs/source/quick_start/md/quick_start_for_heterGraph.md
index 37ac0de4c362a097518f775da340907408f9a31e..aae818c0be633a5708eb2ea29636077f3c919425 100644
--- a/docs/source/quick_start/md/quick_start_for_heterGraph.md
+++ b/docs/source/quick_start/md/quick_start_for_heterGraph.md
@@ -58,8 +58,8 @@ Now, we can build a heterogenous graph by using PGL.
 import paddle.fluid as fluid
 import paddle.fluid.layers as fl
 import pgl
-from pgl.contrib import heter_graph
-from pgl.contrib import heter_graph_wrapper
+from pgl import heter_graph
+from pgl import heter_graph_wrapper
 
 g = heter_graph.HeterGraph(num_nodes=num_nodes,
                             edges=edges,
@@ -160,8 +160,3 @@ for epoch in range(30):
     train_loss = exe.run(fluid.default_main_program(), feed=feed_dict, fetch_list=[loss], return_numpy=True)
     print('Epoch %d | Loss: %f'%(epoch, train_loss[0]))
 ```
-
-
-
-
-