diff --git a/autoload/repeat.vim b/autoload/repeat.vim
index 18b1280..ef72247 100644
--- a/autoload/repeat.vim
+++ b/autoload/repeat.vim
@@ -97,7 +97,7 @@ endfunction
 
 function! repeat#wrap(command,count)
     let preserve = (g:repeat_tick == b:changedtick)
-    exe 'norm! '.(a:count ? a:count : '').a:command . (&foldopen =~# 'undo' ? 'zv' : '')
+    exe 'norm! '.(a:count ? a:count : '').a:command . (&foldopen =~# 'undo\|all' ? 'zv' : '')
     if preserve
         let g:repeat_tick = b:changedtick
     endif