Made dialog tree viewer more robust