diff --git a/chebai/preprocessing/datasets/chebi.py b/chebai/preprocessing/datasets/chebi.py
index 4d77495d..a0af3ac8 100644
--- a/chebai/preprocessing/datasets/chebi.py
+++ b/chebai/preprocessing/datasets/chebi.py
@@ -393,7 +393,7 @@ def _get_data_splits(self) -> tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
             self.test_split,
             self.dynamic_data_split_seed,
         )
-        return splits["train"], splits["val"], splits["test"]
+        return splits["train"], splits["validation"], splits["test"]
 
     def _setup_pruned_test_set(
         self, df_test_chebi_version: pd.DataFrame
diff --git a/pyproject.toml b/pyproject.toml
index beda3a3b..6c00552f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,7 +40,7 @@ dev = [
     "omegaconf",
     "deepsmiles",
     "torchmetrics",
-    "chebi-utils>=0.1.1",
+    "chebi-utils>=0.3",
 ]
 
 linters = [