diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 0c460ad..0a597c2 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,4 +1,4 @@
-default_stages: [ "commit", "commit-msg", "push" ]
+default_stages: [ "pre-commit", "commit-msg", "pre-push" ]
 default_language_version:
   python: python3
 
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index 18c118b..0b69801 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -3,7 +3,7 @@ chex
 colorama
 craftax
 distrax @ git+https://github.com/google-deepmind/distrax  # distrax release doesn't support jax > 0.4.13
-envpool
+envpool; sys_platform == "linux"
 flashbax @ git+https://github.com/instadeepai/flashbax
 flax
 gymnasium