diff --git a/environment.yml b/environment.yml
index 47fe8e4c2a640..ff7c5d56052d2 100644
--- a/environment.yml
+++ b/environment.yml
@@ -19,7 +19,9 @@ dependencies:
   - hypothesis>=3.82
   - isort
   - moto
-  - pytest>=4.0
+  - pycodestyle=2.4
+  - pytest>=4.0.2
+  - pytest-mock
   - sphinx
   - numpydoc
 
diff --git a/requirements-dev.txt b/requirements-dev.txt
index 76aaeefa648f4..02d8b0a70aab6 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -10,7 +10,9 @@ gitpython
 hypothesis>=3.82
 isort
 moto
-pytest>=4.0
+pycodestyle==2.4
+pytest>=4.0.2
+pytest-mock
 sphinx
 numpydoc
 beautifulsoup4>=4.2.1