diff --git a/plugins/vi-mode/vi-mode.plugin.zsh b/plugins/vi-mode/vi-mode.plugin.zsh
index ac9233290607f0010570385c5791fa6eef738654..80df97d6144e1dc97561dc8c3d8906a4b0b82e82 100644
--- a/plugins/vi-mode/vi-mode.plugin.zsh
+++ b/plugins/vi-mode/vi-mode.plugin.zsh
@@ -16,14 +16,7 @@ function zle-keymap-select zle-line-init zle-line-finish {
 
 # Ensure that the prompt is redrawn when the terminal size changes.
 TRAPWINCH() {
-  if [[ -o zle ]]; then
-    # Verify that zle is active before trying to reset-prompt.
-    zle
-    if [[ $? -eq 0 ]]; then
-      zle reset-prompt
-      zle -R
-    fi
-  fi
+  zle && { zle reset-prompt; zle -R }
 }
 
 zle -N zle-line-init