call _p9k_on_expand and subsequenty p10k-on-pre-prompt on every precmd, even when called from zle

1 files changed, 2 insertions, 3 deletions

diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index b51eea47..4c33dad9 100644
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -5810,9 +5810,6 @@ _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
 
@@ -5879,6 +5876,8 @@ _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=''