diff --git a/.cnb/template.yml b/.cnb/template.yml
index e1066af..2b99dfd 100644
--- a/.cnb/template.yml
+++ b/.cnb/template.yml
@@ -58,11 +58,15 @@
   stages:
     - name: '部署k8s模块'
       script:  |
-        echo "${KUBECONFIG_DATA}"  | base64 -d >  ~/.kube/config
-        chmod 600 ~/.kube/config  # 安全性建议
+        echo "${KUBECONFIG_DATA}" | base64 -d > ~/.kube/config
+        chmod 600 ~/.kube/config
+
+        # 如果设置了 KUBE_CONTEXT，则切换上下文
+        if [ -n "${KUBE_CONTEXT}" ]; then
+          kubectl config use-context "${KUBE_CONTEXT}"
+        fi
         export NAMESPACE=${KUBE_NAMESPACE:-default}
-        export DEPLOYMENT_NAME=${KUBE_DEPLOYMENT:? "KUBE_DEPLOYMENT 未设置或者没有默认值"}
-        kubectl rollout restart deployment/${DEPLOYMENT_NAME} -n ${NAMESPACE}
+        kubectl rollout restart deployment/${KUBE_DEPLOYMENT} -n ${NAMESPACE}
 
 .build_images_app_template: &build_images_app_template
   services: