diff --git a/torch_geometric_temporal/nn/attention/tsagcn.py b/torch_geometric_temporal/nn/attention/tsagcn.py
index 71bae2b3..56657631 100644
--- a/torch_geometric_temporal/nn/attention/tsagcn.py
+++ b/torch_geometric_temporal/nn/attention/tsagcn.py
@@ -3,7 +3,7 @@
 import numpy as np
 import torch.nn as nn
 from torch.autograd import Variable
-from torch_geometric.utils.to_dense_adj import to_dense_adj
+from torch_geometric.utils._to_dense_adj import to_dense_adj
 import torch.nn.functional as F