diff --git a/pyproject.toml b/pyproject.toml
index a60829d..3c53e1a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -30,8 +30,8 @@ dependencies = [
     "accelerate",
     "numpy",
     "Pillow",
-    "pybsm>=0.6",
-    "scikit-learn==1.5.2",
+    "pybsm>=0.6,<=0.9.0",
+    "scikit-learn>=1.6.0",
     "smqtk_image_io",
     "tabulate",
     "timm>=1.0.3",