diff --git a/TTS/.models.json b/TTS/.models.json
index 0fb187a4c4..440ca7a9f7 100644
--- a/TTS/.models.json
+++ b/TTS/.models.json
@@ -76,12 +76,12 @@
                 }
             }
         },
-        "zh":{
+        "zh-CN":{
             "baker":{
                 "tacotron2-DDC-GST":{
-                    "model_file": "1RR9rZdV_FMm8yvtCHALtUbJf1nxbUiAw",
-                    "config_file": "1daY1JHGXEozJ-MGYLiWEUmzEwEvM5xpz",
-                    "stats_file": "1vl9c-D3dW_E7pdhNpDFQLX-giJc0jOtV",
+                    "model_file": "1SYpv7V__QYDjKXa_vJmNXo1CSkcoZovy",
+                    "config_file": "14BIvfJXnFHi3jcxYNX40__TR6RwJOZqi",
+                    "stats_file": "1ECRlXybT6rAWp269CkhjUPwcZ10CkcqD",
                     "commit": ""
                 }
             }
diff --git a/TTS/tts/utils/text/__init__.py b/TTS/tts/utils/text/__init__.py
index 4f4a38eaf5..2a724650f1 100644
--- a/TTS/tts/utils/text/__init__.py
+++ b/TTS/tts/utils/text/__init__.py
@@ -41,7 +41,7 @@ def text2phone(text, language):
     '''
 
     # TO REVIEW : How to have a good implementation for this?
-    if language == "chinese-mandarin":
+    if language == "zh-CN":
         ph = chinese_text_to_phonemes(text)
         return ph