diff --git a/python/dgl/batch.py b/python/dgl/batch.py
index b97b925d463b..a0a9430721ea 100644
--- a/python/dgl/batch.py
+++ b/python/dgl/batch.py
@@ -428,4 +428,4 @@ def unbatch_hetero(*args, **kwargs):
     """DEPREACTED: please use dgl.unbatch """
     dgl_warning('From v0.5, DGLHeteroGraph is merged into DGLGraph. You can safely'
                 ' replace dgl.unbatch_hetero with dgl.unbatch')
-    return batch(*args, **kwargs)
+    return unbatch(*args, **kwargs)