diff --git a/matdeeplearn/models/torchmd_etEarly.py b/matdeeplearn/models/torchmd_etEarly.py
index 7691a386..2edf74a8 100644
--- a/matdeeplearn/models/torchmd_etEarly.py
+++ b/matdeeplearn/models/torchmd_etEarly.py
@@ -4,7 +4,7 @@
 from torch import Tensor, nn
 import torch_geometric.nn
 from torch_geometric.nn import MessagePassing
-from torch_scatter import scatter
+from torch_geometric.utils import scatter
 from matdeeplearn.models.utils import (
     NeighborEmbedding,
     CosineCutoff,