diff --git a/topobench/data/loaders/graph/adme_datasets.py b/topobench/data/loaders/graph/adme_datasets.py
index 91daa52e8..c3653be3e 100644
--- a/topobench/data/loaders/graph/adme_datasets.py
+++ b/topobench/data/loaders/graph/adme_datasets.py
@@ -4,7 +4,7 @@
 from pathlib import Path
 
 import torch
-from ogb.utils import smiles2graph
+from ogb.utils.mol import smiles2graph
 from omegaconf import DictConfig
 from tdc.single_pred import ADME
 from torch_geometric.data import Data, InMemoryDataset