diff --git a/environment.yml b/environment.yml
index 9c2edb8..58c7a71 100644
--- a/environment.yml
+++ b/environment.yml
@@ -14,6 +14,6 @@ dependencies:
   - scanpy
   - squidpy
   - tensorflow=2.13
-  - tensorflow-probability=0.22.1
+  - tensorflow-probability=0.22
   - pip:
       - -e .