diff --git a/scripts/helm/upgrade.sh b/scripts/helm/upgrade.sh
index f57d2fb07..3c437db9a 100644
--- a/scripts/helm/upgrade.sh
+++ b/scripts/helm/upgrade.sh
@@ -81,6 +81,7 @@ patch
 installation_type=1
 if [[ ${ENTERPRISE} -eq 1 ]]; then
     cp -rf ../../ee/scripts/* ../../scripts/
+    sed 's/\(image_tag.*[0-9]\)\(-pr\)\?"$/\1\2-ee"/' vars.yaml
     echo -e "Migrating clickhouse"
     migration clickhouse
 fi