diff --git a/pyproject.toml b/pyproject.toml
index 0134a30..6ab06a5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ description = "Fine-tuning pipeline to distill floop's LLM consolidator into a s
 requires-python = ">=3.11"
 dependencies = [
     "pyyaml>=6.0",
-    "datasets>=2.0",
+    "datasets>=4.8.4",
 ]
 
 [project.optional-dependencies]