diff --git a/environment.yml b/environment.yml
index fa34d9433..0ff1e0a2d 100644
--- a/environment.yml
+++ b/environment.yml
@@ -52,7 +52,7 @@ dependencies:
   # Control blas/openmp threads
   - threadpoolctl
   - pip:
-    - git+https://github.com/OpenFreeEnergy/gufe@main
+    - git+https://github.com/OpenFreeEnergy/gufe@maint/add_py_typed
   - run_constrained:
     # drop this pin when handled upstream in espaloma-feedstock
     - smirnoff99frosst>=1.1.0.1  #https://github.com/openforcefield/smirnoff99Frosst/issues/109