diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index 4c33dad9..b51eea47 100644
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -5810,6 +5810,9 @@ _p9k_precmd_impl() {
     if zle; then
       __p9k_new_status=0
       __p9k_new_pipestatus=(0)
+      _p9k__expanded=1
+    else
+      _p9k__expanded=0
       _p9k__must_restore_prompt=0
     fi
 
@@ -5876,8 +5879,6 @@ _p9k_precmd_impl() {
     _p9k_worker_invoke ${f_compute%% *} ${(e)f_compute}
   done
 
-  _p9k__expanded=0
-
   _p9k__refresh_reason=precmd
   _p9k_set_prompt
   _p9k__refresh_reason=''