diff --git a/src/python/piper_train/vits/mel_processing.py b/src/python/piper_train/vits/mel_processing.py
index 1776f50..72e81e2 100644
--- a/src/python/piper_train/vits/mel_processing.py
+++ b/src/python/piper_train/vits/mel_processing.py
@@ -103,7 +103,9 @@ def mel_spectrogram_torch(
     fmax_dtype_device = str(fmax) + "_" + dtype_device
     wnsize_dtype_device = str(win_size) + "_" + dtype_device
     if fmax_dtype_device not in mel_basis:
-        mel = librosa_mel_fn(sampling_rate, n_fft, num_mels, fmin, fmax)
+        mel = librosa_mel_fn(
+            sr=sampling_rate, n_fft=n_fft, n_mels=num_mels, fmin=fmin, fmax=fmax
+        )
         mel_basis[fmax_dtype_device] = torch.from_numpy(mel).type_as(y)
     if wnsize_dtype_device not in hann_window:
         hann_window[wnsize_dtype_device] = torch.hann_window(win_size).type_as(y)