1 files changed, 8 insertions, 3 deletions
diff --git a/urgent.tcl b/urgent.tcl
index 4242987..b886eb5 100644
--- a/urgent.tcl
+++ b/urgent.tcl
@@ -104,13 +104,17 @@ proc urgent::chat_xwinid {chatid} {
     xclientwinid $tkwin
 }
 
+proc urgent::initialize {chatid type} {
+    variable state
+    set state(active,$chatid) false
+}
+
 proc urgent::record_xwinid {chatid _type} {
     variable state
     set state(xwinid,$chatid) [xwinid $chatid]
-    set state(active,$chatid) false
 }
 
-proc urgent::forget_xwinid {chatid} {
+proc urgent::cleanup {chatid} {
     variable state
 
     unset state(xwinid,$chatid)
@@ -165,8 +169,9 @@ namespace eval urgent {
 	}
     }]
 
+    hook::add open_chat_pre_hook [namespace current]::initialize
     hook::add open_chat_post_hook [namespace current]::record_xwinid 40
-    hook::add close_chat_post_hook [namespace current]::forget_xwinid
+    hook::add close_chat_post_hook [namespace current]::cleanup
     hook::add draw_message_hook [namespace current]::chat_message_notify 19
     hook::add got_focus_hook [namespace current]::clear_urgency_hint
 }