diff --git a/src/graphnet/models/utils.py b/src/graphnet/models/utils.py
index e1ef7956c..1d77903a5 100644
--- a/src/graphnet/models/utils.py
+++ b/src/graphnet/models/utils.py
@@ -6,7 +6,7 @@
 import torch
 from torch import Tensor, LongTensor
 
-from torch_geometric.utils.homophily import homophily
+from torch_geometric.utils import homophily
 
 
 def calculate_xyzt_homophily(