diff --git a/data_utils.py b/data_utils.py
index 3a369e8..f6ebfe8 100644
--- a/data_utils.py
+++ b/data_utils.py
@@ -135,10 +135,10 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
             bert = torch.load(bert_path)
             assert bert.shape[-1] == len(phone)
         except:
-            #bert = get_bert(text, word2ph, language_str)
-            #torch.save(bert, bert_path)
-            print(bert.shape[-1], bert_path, text, pold)
-            #assert bert.shape[-1] == len(phone)
+            bert = get_bert(text, word2ph, language_str)
+            torch.save(bert, bert_path)
+            #print(bert.shape[-1], bert_path, text, pold)
+            assert bert.shape[-1] == len(phone)
 
         assert bert.shape[-1] == len(phone), (
         bert.shape, len(phone), sum(word2ph), p1, p2, t1, t2, pold, pold2, word2ph, text, w2pho)