diff --git a/examples/pytorch/hgp_sl/functions.py b/examples/pytorch/hgp_sl/functions.py
index 3c22261f2e6b..c12d72d25a8f 100644
--- a/examples/pytorch/hgp_sl/functions.py
+++ b/examples/pytorch/hgp_sl/functions.py
@@ -9,10 +9,10 @@
 """
 import dgl
 import torch
+from dgl._sparse_ops import _gsddmm, _gspmm
 from dgl.backend import astype
 from dgl.base import ALL, is_all
 from dgl.heterograph_index import HeteroGraphIndex
-from dgl.sparse import _gsddmm, _gspmm
 from torch import Tensor
 from torch.autograd import Function