diff --git a/chebai_graph/preprocessing/datasets/pubchem.py b/chebai_graph/preprocessing/datasets/pubchem.py
index 6f5d118..716491f 100644
--- a/chebai_graph/preprocessing/datasets/pubchem.py
+++ b/chebai_graph/preprocessing/datasets/pubchem.py
@@ -1,7 +1,7 @@
-from chebai.preprocessing.datasets.pubchem import PubchemChem
+from chebai.preprocessing.datasets.pubchem import PubChem
 
 from chebai_graph.preprocessing.datasets.chebi import GraphPropertiesMixIn
 
 
-class PubChemGraphProperties(GraphPropertiesMixIn, PubchemChem):
+class PubChemGraphProperties(GraphPropertiesMixIn, PubChem):
     pass