diff --git a/pyproject.toml b/pyproject.toml
index 1c3e25b..9a8642f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -60,7 +60,7 @@ dev = [
     "ruff",
 ]
 
-distill = ["torch", "transformers<5.4.0", "scikit-learn", "skeletoken>=0.3.2"]
+distill = ["torch", "transformers<5.4.0", "scikit-learn", "skeletoken>=0.3.3"]
 onnx = ["onnx", "torch"]
 # train also installs inference
 train = ["torch", "lightning", "scikit-learn", "skops"]