diff --git a/training/heterogeneous-clusters/tf.data.service.sagemaker/code/requirements.txt b/training/heterogeneous-clusters/tf.data.service.sagemaker/code/requirements.txt
index 64b956a22e..10aac59994 100644
--- a/training/heterogeneous-clusters/tf.data.service.sagemaker/code/requirements.txt
+++ b/training/heterogeneous-clusters/tf.data.service.sagemaker/code/requirements.txt
@@ -1,2 +1,2 @@
-protobuf==3.20.1
+protobuf==3.20.2
 tensorflow-addons==0.17.0
\ No newline at end of file