From 4070471f0f227f0887ee1eaed209ff3a917a28ea Mon Sep 17 00:00:00 2001
From: Jay Lee <jay0lee@gmail.com>
Date: Mon, 23 Jun 2014 13:32:30 -0400
Subject: [PATCH] more cleanup on mbox-restore label changes

---
 gyb.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/gyb.py b/gyb.py
index e3787df..7f46a48 100644
--- a/gyb.py
+++ b/gyb.py
@@ -999,8 +999,11 @@ def main(argv):
             if bad_label in labels:
               labels.remove(bad_label)
               if bad_label == u'Chat':
-                bad_label = u'Restored Chat'
-              labels.append(u'\\\\%s' % bad_label)
+                labels.append(u'Restored Chats')
+              elif bad_label == u'Drafts':
+                labels.append(u'\\\\Draft')
+              else:
+                labels.append(u'\\\\%s' % bad_label)
           escaped_labels = []
           for label in labels:
             if label.find('\"') != -1: