diff --git a/scripts/helm/kube-install.sh b/scripts/helm/kube-install.sh
index 0a42416bf..056a45de4 100755
--- a/scripts/helm/kube-install.sh
+++ b/scripts/helm/kube-install.sh
@@ -147,7 +147,7 @@ function enterprise(){
     enterprise=1
     sed -i "s#enterprise_edition_license.*#enterprise_edition_license: \"${1}\"#g" vars.yaml
     # Updating image version to be ee
-    sed -i "s/\(image_tag.*[0-9]\)\"$/\1-ee\"/" vars.yaml
+    sed 's/\(image_tag.*[0-9]\)\(-pr\)\?"$/\1\2-ee"/' vars.yaml
     echo "Importing enterprise code..."
     cp -rf ../../ee/scripts/* ../
 }