diff --git a/autoload/vimtex/fold.vim b/autoload/vimtex/fold.vim
index f28c0a8416..89f2d7b6ca 100644
--- a/autoload/vimtex/fold.vim
+++ b/autoload/vimtex/fold.vim
@@ -85,9 +85,13 @@ endfunction
 " }}}1
 
 function! vimtex#fold#refresh(map) abort " {{{1
-  setlocal foldmethod=expr
-  execute 'normal!' a:map
-  setlocal foldmethod=manual
+  if &diff
+    setlocal foldmethod=diff
+  else
+    setlocal foldmethod=expr
+    execute 'normal!' a:map
+    setlocal foldmethod=manual
+  endif
 endfunction
 
 " }}}1