diff --git a/environment.yml b/environment.yml
index 019fcf7..e74c686 100644
--- a/environment.yml
+++ b/environment.yml
@@ -4,7 +4,7 @@ channels:
   - conda-forge
 
 dependencies:
-  - python>=3.10
+  - python=3.10
   - anndata
   - click
   - matplotlib
@@ -14,6 +14,6 @@ dependencies:
   - scanpy
   - squidpy
   - tensorflow>=2.11
-  - tensorflow-probability>=0.19
+  - tensorflow-probability=0.2
   - pip:
       - -e .