diff --git a/distribution_shift_framework/requirements.txt b/distribution_shift_framework/requirements.txt
index 46c3735..a788a0e 100644
--- a/distribution_shift_framework/requirements.txt
+++ b/distribution_shift_framework/requirements.txt
@@ -34,7 +34,7 @@ opt-einsum==3.3.0
 optax==0.1.1
 pkg_resources==0.0.0
 promise==2.3
-protobuf==3.19.4
+protobuf==4.25.8
 pyasn1==0.4.8
 pyasn1-modules==0.2.8
 PyYAML==6.0