diff --git a/serving/docker/scripts/security_patch.sh b/serving/docker/scripts/security_patch.sh
index 52b8145df..96f5373bd 100755
--- a/serving/docker/scripts/security_patch.sh
+++ b/serving/docker/scripts/security_patch.sh
@@ -4,7 +4,8 @@ IMAGE_NAME=$1
 
 apt-get update
 
-if [[ "$IMAGE_NAME" == "deepspeed" ]]; then
+if [[ "$IMAGE_NAME" == "deepspeed" ]] || \
+   [[ "$IMAGE_NAME" == "pytorch-gpu" ]]; then
   apt-get upgrade -y dpkg openssl curl libssl3
 elif [[ "$IMAGE_NAME" == "cpu" ]]; then
   apt-get upgrade -y libpcre2-8-0 libdbus-1-3 curl