diff --git a/environment.yml b/environment.yml
index c841d5f..019fcf7 100644
--- a/environment.yml
+++ b/environment.yml
@@ -14,6 +14,6 @@ dependencies:
   - scanpy
   - squidpy
   - tensorflow>=2.11
-  - tensorflow-probability=0.19
+  - tensorflow-probability>=0.19
   - pip:
       - -e .