diff --git a/autoenv/autoenv.zsh b/autoenv/autoenv.zsh
index 7bb0e79..48beefe 100644
--- a/autoenv/autoenv.zsh
+++ b/autoenv/autoenv.zsh
@@ -190,6 +190,7 @@ _autoenv_snap_restore() {
   unset "_autoenv_snap_pre_path[$dir]" "_autoenv_snap_pre_ps1[$dir]" \
         "_autoenv_snap_diff_funcs[$dir]" "_autoenv_snap_diff_aliases[$dir]" \
         "_autoenv_snap_diff_env[$dir]"
+  prompt "${prompt_theme[@]}"
 }
 
 # Load zstat from stat module for inspecting modified time.
diff --git a/prompt_fresh_setup b/prompt_fresh_setup
index a7ba1c4..25f0c93 100644
--- a/prompt_fresh_setup
+++ b/prompt_fresh_setup
@@ -144,8 +144,12 @@ fresh_line_one() {
     local docker=" %{%F{6}%}$DOCKER_MACHINE_NAME%{%f%}"
   fi
 
+  if [[ ! -z "$MODULAR_PATH" ]]; then
+    local modular=" 🔥"
+  fi
+
   # Print the first line of the prompt
-  print -P "$time_stamp $directory$git$py$docker$result"
+  print -P "$time_stamp $directory$git$py$docker$modular$result"
 }
 
 # Executed before each prompt.