diff --git a/dataflow/gemma-flex-template/requirements-test.txt b/dataflow/gemma-flex-template/requirements-test.txt
index 5e6dcfc99aa..bd6055956ce 100644
--- a/dataflow/gemma-flex-template/requirements-test.txt
+++ b/dataflow/gemma-flex-template/requirements-test.txt
@@ -2,6 +2,6 @@ google-cloud-aiplatform==1.62.0
 google-cloud-dataflow-client==0.8.14
 google-cloud-pubsub==2.28.0
 google-cloud-storage==2.18.2
-pytest==8.3.2
+pytest==9.0.3
 pytest-timeout==2.3.1
 pyyaml==6.0.2