diff --git a/people-and-planet-ai/weather-forecasting/tests/dataset_tests/requirements.txt b/people-and-planet-ai/weather-forecasting/tests/dataset_tests/requirements.txt
index b6ae8de9dbb..66292e154dc 100644
--- a/people-and-planet-ai/weather-forecasting/tests/dataset_tests/requirements.txt
+++ b/people-and-planet-ai/weather-forecasting/tests/dataset_tests/requirements.txt
@@ -1,4 +1,4 @@
 ../../serving/weather-data
 apache-beam[gcp,interactive]==2.43.0
-build==0.10.0
+build==1.6.1
 plotly==5.15.0
diff --git a/people-and-planet-ai/weather-forecasting/tests/training_tests/requirements.txt b/people-and-planet-ai/weather-forecasting/tests/training_tests/requirements.txt
index d2b1b4ba08d..63e0c3ea753 100644
--- a/people-and-planet-ai/weather-forecasting/tests/training_tests/requirements.txt
+++ b/people-and-planet-ai/weather-forecasting/tests/training_tests/requirements.txt
@@ -1,3 +1,3 @@
 ../../serving/weather-model
-build==0.10.0
+build==1.6.1
 google-cloud-aiplatform==1.157.0