diff-hl-margin-mode: Walk buffers outside of the if form

diff --git a/diff-hl-margin.el b/diff-hl-margin.el
index 57a0b703bc894e10067f89514a192dd70204aab9..6745742c543b7c9f1984091471c8ca6d834dd3bf 100644 (file)
--- a/diff-hl-margin.el
+++ b/diff-hl-margin.el
@@ -45,24 +45,17 @@
       (progn
         (setq diff-hl-margin-old-highlight-function diff-hl-highlight-function
               diff-hl-highlight-function 'diff-hl-highlight-on-margin)
-        (setq-default left-margin-width 1)
-        (dolist (buffer (buffer-list))
-          (with-current-buffer buffer
-            (cond
-             (diff-hl-mode
-              (diff-hl-update))
-             (diff-hl-dired-mode
-              (diff-hl-dired-update))))))
+        (setq-default left-margin-width 1))
     (setq diff-hl-highlight-function diff-hl-margin-old-highlight-function
           diff-hl-margin-old-highlight-function nil)
-    (setq-default left-margin-width 0)
-    (dolist (buffer (buffer-list))
-      (with-current-buffer buffer
-        (cond
-         (diff-hl-mode
-          (diff-hl-update))
-         (diff-hl-dired-mode
-          (diff-hl-dired-update))))))
+    (setq-default left-margin-width 0))
+  (dolist (buffer (buffer-list))
+    (with-current-buffer buffer
+      (cond
+       (diff-hl-mode
+        (diff-hl-update))
+       (diff-hl-dired-mode
+        (diff-hl-dired-update)))))
   (walk-windows (lambda (win) (set-window-buffer win (window-buffer win)))))
 
 (defvar diff-hl-margin-spec-cache