Add assoc_with_msg method for better dialog handling.