diff --git a/TTS/.models.json b/TTS/.models.json
index 0fb187a4..440ca7a9 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 4f4a38ea..2a724650 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
 
diff --git a/notebooks/Chinese_Mandarin_DDC_GST_Tacotron2_TTS_and_MultiBand_MelGAN_Example.ipynb b/notebooks/Chinese_Mandarin_DDC_GST_Tacotron2_TTS_and_MultiBand_MelGAN_Example.ipynb
index 709dbb8d..b28dd90c 100644
--- a/notebooks/Chinese_Mandarin_DDC_GST_Tacotron2_TTS_and_MultiBand_MelGAN_Example.ipynb
+++ b/notebooks/Chinese_Mandarin_DDC_GST_Tacotron2_TTS_and_MultiBand_MelGAN_Example.ipynb
@@ -64,9 +64,9 @@
    ],
    "source": [
     "! mkdir data/\n",
-    "! gdown --id 1RR9rZdV_FMm8yvtCHALtUbJf1nxbUiAw -O data/tts_model.pth.tar\n",
-    "! gdown --id 1daY1JHGXEozJ-MGYLiWEUmzEwEvM5xpz -O data/tts_config.json\n",
-    "! gdown --id 1vl9c-D3dW_E7pdhNpDFQLX-giJc0jOtV -O data/tts_scale_stats.npy"
+    "! gdown --id 1SYpv7V__QYDjKXa_vJmNXo1CSkcoZovy -O data/tts_model.pth.tar\n",
+    "! gdown --id 14BIvfJXnFHi3jcxYNX40__TR6RwJOZqi -O data/tts_config.json\n",
+    "! gdown --id 1ECRlXybT6rAWp269CkhjUPwcZ10CkcqD -O data/tts_scale_stats.npy"
    ]
   },
   {
@@ -526,4 +526,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}
\ No newline at end of file