diff --git a/train.py b/train.py
index 3e7ad0ad..cd969e9d 100644
--- a/train.py
+++ b/train.py
@@ -47,7 +47,9 @@ def setup_loader(is_val=False):
             min_seq_len=0 if is_val else c.min_seq_len,
             max_seq_len=float("inf") if is_val else c.max_seq_len,
             cached=False if c.dataset != "tts_cache" else True,
-            phoneme_cache_path=c.phoneme_cache_path
+            phoneme_cache_path=c.phoneme_cache_path,
+            use_phonemes=c.use_phonemes,
+            phoneme_language=c.phoneme_language
             )
         loader = DataLoader(
             dataset,