diff --git a/prompt_fresh_setup b/prompt_fresh_setup
index 7a139f2..dada470 100644
--- a/prompt_fresh_setup
+++ b/prompt_fresh_setup
@@ -44,7 +44,7 @@ prompt_fresh_setup() {
   fi
 
   local userhost=$USER
-  if [ "$SSH_CONNECTION" != "" ] || [ "$DISTROBOX_ENTER_PATH" != "" ]; then
+  if [ "$SSH_CONNECTION" != "" ] || [ "$container" != "" ]; then
     local user="$user@%{%F{244}%}%M%{%f%}"
     local userhost="$userhost@`hostname`"
   fi