diff --git a/espaloma/nn/sequential.py b/espaloma/nn/sequential.py
index 4f1ef620..53c81f05 100644
--- a/espaloma/nn/sequential.py
+++ b/espaloma/nn/sequential.py
@@ -135,7 +135,7 @@ def forward(self, g, x=None):
         import dgl
 
         # get homogeneous subgraph
-        g_ = dgl.to_homo(g.edge_type_subgraph(["n1_neighbors_n1"]))
+        g_ = dgl.to_homogeneous(g.edge_type_subgraph(["n1_neighbors_n1"]))
 
         if x is None:
             # get node attributes