diff --git a/TTS/tts/datasets/dataset.py b/TTS/tts/datasets/dataset.py
index 6d177743..513f2b12 100644
--- a/TTS/tts/datasets/dataset.py
+++ b/TTS/tts/datasets/dataset.py
@@ -273,6 +273,7 @@ class TTSDataset(Dataset):
         item = args[0]
         func_args = args[1]
         text, wav_file, *_ = item
+        func_args[3] = item[4]
         phonemes = TTSDataset._load_or_generate_phoneme_sequence(wav_file, text, *func_args)
         return phonemes