diff --git a/pyproject.toml b/pyproject.toml
index 0a58cc61..0be3a158 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -30,7 +30,7 @@ dependencies = [
     "numpy",
     "scikit-image",
     "scikit-learn",
-    "keras>=3.5.0",
+    "keras==3.5.0",
     "torch>=2.1.0",
     "tifffile",
     "tqdm",