Fixed.

[riece] / lisp / riece-toolbar.el

diff --git a/lisp/riece-toolbar.el b/lisp/riece-toolbar.el
index 1335eca..4bcec8a 100644 (file)
--- a/lisp/riece-toolbar.el
+++ b/lisp/riece-toolbar.el
@@ -86,8 +86,10 @@
                  (specifier-specs default-toolbar (current-buffer)))
            (set-specifier default-toolbar toolbar (current-buffer)))
          (defun riece-unset-toolbar ()
-           (set-specifier default-toolbar riece-toolbar-original-toolbar
-                          (current-buffer))
+           (if riece-toolbar-original-toolbar
+               (set-specifier default-toolbar riece-toolbar-original-toolbar
+                              (current-buffer))
+             (remove-specifier default-toolbar (current-buffer)))
            (kill-local-variable 'riece-toolbar-original-toolbar)))
       (defalias 'riece-make-toolbar-from-menu 'ignore)
       (defalias 'riece-set-toolbar 'ignore)
@@ -119,15 +121,18 @@
   '(riece-menu))
 
 (defun riece-toolbar-insinuate ()
+  (if riece-command-buffer
+      (with-current-buffer riece-command-buffer
+       (riece-toolbar-command-mode-hook)))
   (add-hook 'riece-command-mode-hook
-           'riece-toolbar-command-mode-hook
-           t))
+           'riece-toolbar-command-mode-hook t))
 
 (defun riece-toolbar-uninstall ()
+  (if riece-command-buffer
+      (with-current-buffer riece-command-buffer
+       (riece-unset-toolbar)))
   (remove-hook 'riece-command-mode-hook
-              'riece-toolbar-command-mode-hook)
-  (with-current-buffer riece-command-buffer
-    (riece-unset-toolbar)))
+              'riece-toolbar-command-mode-hook))
 
 (provide 'riece-toolbar)