fix toggletag command

1 files changed, 4 insertions, 3 deletions

diff --git a/alot/commands.py b/alot/commands.py
index a78c2579..e6d0c6e3 100644
--- a/alot/commands.py
+++ b/alot/commands.py
@@ -293,13 +293,15 @@ class FlushCommand(Command):
 class ToggleThreadTagCommand(Command):
     """
     """
-    def __init__(self, thread, tag, **kwargs):
-        assert thread
+    def __init__(self, tag, thread=None, **kwargs):
+        assert tag
         self.thread = thread
         self.tag = tag
         Command.__init__(self, **kwargs)
 
     def apply(self, ui):
+        cb = ui.current_buffer
+        self.thread = cb.get_selected_thread()
         try:
             if self.tag in self.thread.get_tags():
                 self.thread.remove_tags([self.tag])
@@ -314,7 +316,6 @@ class ToggleThreadTagCommand(Command):
 
         # update current buffer
         # TODO: what if changes not yet flushed?
-        cb = ui.current_buffer
         if isinstance(cb, buffer.SearchBuffer):
             # refresh selected threadline
             threadwidget = cb.get_selected_threadline()