diff --git a/environment.yml b/environment.yml
index 79f3e6e..f7e5043 100644
--- a/environment.yml
+++ b/environment.yml
@@ -35,5 +35,5 @@ dependencies:
   - conda-forge::rdkit
   - conda-forge::ipdb
   - transformers
-  - dgllife
-  - dglteam=2.0.0
+  - conda-forge::dgllife
+  - dglteam::dgl<=2.0