diff --git a/bin/load-kyuubi-env.sh b/bin/load-kyuubi-env.sh
index fdff1e0c571..779ac2bea5a 100755
--- a/bin/load-kyuubi-env.sh
+++ b/bin/load-kyuubi-env.sh
@@ -16,10 +16,7 @@
 # limitations under the License.
 #
 
-
-if [ -z "${KYUUBI_HOME}" ]; then
-  export KYUUBI_HOME="$(cd "$(dirname "$0")"/.. || exit; pwd)"
-fi
+export KYUUBI_HOME="${KYUUBI_HOME:-"$(cd "$(dirname "$0")"/.. || exit; pwd)"}"
 export KYUUBI_CONF_DIR="${KYUUBI_CONF_DIR:-"${KYUUBI_HOME}"/conf}"
 
 silent=0