diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index 1e01ba5..9668c42 100644
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -8280,7 +8280,7 @@ _p9k_init_toolbox() {
     local name=(${(Q)${${(@M)${(f)"$(</run/.containerenv)"}:#name=*}#name=}})
     [[ ${#name} -eq 1 && -n ${name[1]} ]] || return 0
     typeset -g P9K_TOOLBOX_NAME=${name[1]}
-  elif [[ -n $DISTROBOX_ENTER_PATH && -n $NAME ]]; then
+  elif [[ -n $DISTROBOX_ENTER_PATH && -n $HOST ]]; then
     local name=${(%):-%m}
     if [[ $name == $NAME* ]]; then
       typeset -g P9K_TOOLBOX_NAME=$name