From be23742305bfe1f841d13fcbff81e646c135dce4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Hern=C3=A1n=20Cervera?= <hjcerveram@hotmail.com>
Date: Sat, 4 Apr 2026 13:52:30 -0600
Subject: [PATCH] fix: visual mode highlight lingers after put

---
 zsh-vi-mode.zsh | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/zsh-vi-mode.zsh b/zsh-vi-mode.zsh
index c3d40f1..5ec5a17 100644
--- a/zsh-vi-mode.zsh
+++ b/zsh-vi-mode.zsh
@@ -1152,7 +1152,6 @@ function zvm_vi_put_after() {
   local count=${NUMERIC:-1}
   local head= foot=
   local content=${CUTBUFFER}
-  local offset=1
 
   if [[ ${content: -1} == $'\n' ]]; then
     local pos=${CURSOR}
@@ -1183,7 +1182,6 @@ function zvm_vi_put_after() {
       repeated+="$content"
     done
 
-    offset=0
     BUFFER="${head}${repeated}${foot}"
     CURSOR=$pos
   else
@@ -1207,9 +1205,8 @@ function zvm_vi_put_after() {
     CURSOR=$CURSOR+$#repeated
   fi
 
-  # Refresh display and highlight buffer
+  # Refresh display
   zvm_highlight clear
-  zvm_highlight custom $(($#head+$offset)) $(($#head+$#repeated+$offset))
 }
 
 # Put cutbuffer before the cursor
@@ -1260,9 +1257,8 @@ function zvm_vi_put_before() {
     CURSOR=$((CURSOR-1))
   fi
 
-  # Refresh display and highlight buffer
+  # Refresh display
   zvm_highlight clear
-  zvm_highlight custom $#head $(($#head+$#repeated))
 }
 
 # Replace a selection