From e9b0809d32fea1079406d948cfe6a2f146b87f49 Mon Sep 17 00:00:00 2001
From: Rajesh Rajendran <rjshrjndrn@gmail.com>
Date: Thu, 13 Jan 2022 16:31:26 +0530
Subject: [PATCH] chore(upgrade): inject ee image

Signed-off-by: Rajesh Rajendran <rjshrjndrn@gmail.com>
---
 scripts/helm/upgrade.sh | 1 +
 1 file changed, 1 insertion(+)

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