diff --git a/code_examples/tensorflow2/abc_covid_19/requirements.txt b/code_examples/tensorflow2/abc_covid_19/requirements.txt
index e0c771dc5..d374e83b9 100644
--- a/code_examples/tensorflow2/abc_covid_19/requirements.txt
+++ b/code_examples/tensorflow2/abc_covid_19/requirements.txt
@@ -1,4 +1,4 @@
 tensorflow-probability==0.12.2
 pytest==6.2.5
 pytest-pythonpath==0.7.3
-pytest-mock==2.0.0
+pytest-mock==3.7.0
diff --git a/code_examples/tensorflow2/adversarial_generalized_method_of_moments/requirements.txt b/code_examples/tensorflow2/adversarial_generalized_method_of_moments/requirements.txt
index 26d241363..a83152617 100644
--- a/code_examples/tensorflow2/adversarial_generalized_method_of_moments/requirements.txt
+++ b/code_examples/tensorflow2/adversarial_generalized_method_of_moments/requirements.txt
@@ -3,4 +3,4 @@ sklearn
 pyyaml
 pytest==6.2.5
 pytest-pythonpath==0.7.3
-pytest-mock==2.0.0
+pytest-mock==3.7.0