diff --git a/shared/envs/kipoi-py3-keras1.2.yaml b/shared/envs/kipoi-py3-keras1.2.yaml
index f6b5f2103..f030edea6 100644
--- a/shared/envs/kipoi-py3-keras1.2.yaml
+++ b/shared/envs/kipoi-py3-keras1.2.yaml
@@ -11,7 +11,7 @@ dependencies:
 - pysam
 - python=3.6
 - h5py
-- genomelake==0.1.4
+- genomelake=0.1.4
 - numpy
 - pandas
 - cython
@@ -26,3 +26,4 @@ dependencies:
   - deepcpg==1.0.4
   # other 
   - ipykernel
+  - typing-extensions==4.1.1
\ No newline at end of file
diff --git a/shared/envs/kipoi-py3-keras2.yaml b/shared/envs/kipoi-py3-keras2.yaml
index 2ebb37e9b..fe3bd7899 100644
--- a/shared/envs/kipoi-py3-keras2.yaml
+++ b/shared/envs/kipoi-py3-keras2.yaml
@@ -12,7 +12,7 @@ dependencies:
 - bedtools
 - pybedtools
 - pysam
-- genomelake==0.1.4
+- genomelake=0.1.4
 - cython
 - h5py
 - pytorch-cpu>=0.2.0
@@ -23,7 +23,7 @@ dependencies:
 - ls-gkm=0.0.1
 - pybigwig
 - gtfparse>=1.0.7
-- scikit-learn==0.18.1
+- scikit-learn=0.18.1
 - pip:
   - kipoi_veff
   - kipoi_interpret
@@ -44,3 +44,4 @@ dependencies:
   - concise>=0.6.6
   # other 
   - ipykernel
+  - typing-extensions==4.1.1