diff --git a/environment.yml b/environment.yml
index 943d4c5..704fb9f 100644
--- a/environment.yml
+++ b/environment.yml
@@ -46,11 +46,11 @@ dependencies:
 
     # Pip-only installs
   - pip:
-    - git+https://github.com/choderalab/asapdiscovery@write_and_plot_splits#egg=asapdiscovery-alchemy&subdirectory=asapdiscovery-alchemy
-    - git+https://github.com/choderalab/asapdiscovery@write_and_plot_splits#egg=asapdiscovery-data&subdirectory=asapdiscovery-data
-    - git+https://github.com/choderalab/asapdiscovery@write_and_plot_splits#egg=asapdiscovery-modeling&subdirectory=asapdiscovery-modeling
-    - git+https://github.com/choderalab/asapdiscovery@write_and_plot_splits#egg=asapdiscovery-docking&subdirectory=asapdiscovery-docking
-    - git+https://github.com/choderalab/asapdiscovery@write_and_plot_splits#egg=asapdiscovery-ml&subdirectory=asapdiscovery-ml
+    - git+https://github.com/choderalab/asapdiscovery@main#egg=asapdiscovery-alchemy&subdirectory=asapdiscovery-alchemy
+    - git+https://github.com/choderalab/asapdiscovery@main#egg=asapdiscovery-data&subdirectory=asapdiscovery-data
+    - git+https://github.com/choderalab/asapdiscovery@main#egg=asapdiscovery-modeling&subdirectory=asapdiscovery-modeling
+    - git+https://github.com/choderalab/asapdiscovery@main#egg=asapdiscovery-docking&subdirectory=asapdiscovery-docking
+    - git+https://github.com/choderalab/asapdiscovery@main#egg=asapdiscovery-ml&subdirectory=asapdiscovery-ml
     - streamlit-ketcher