diff --git a/mozilla_voice_tts/tts/datasets/TTSDataset.py b/mozilla_voice_tts/tts/datasets/TTSDataset.py
index 2ef78e11..1ecca75f 100644
--- a/mozilla_voice_tts/tts/datasets/TTSDataset.py
+++ b/mozilla_voice_tts/tts/datasets/TTSDataset.py
@@ -72,7 +72,7 @@ class MyDataset(Dataset):
         self.sort_items()
 
     def load_wav(self, filename):
-        audio = self.ap.load_wav(filename, sr=self.sample_rate)
+        audio = self.ap.load_wav(filename)
         return audio
 
     @staticmethod