diff --git a/rvc/f0/mel.py b/rvc/f0/mel.py
index 439a258..2c06800 100644
--- a/rvc/f0/mel.py
+++ b/rvc/f0/mel.py
@@ -19,7 +19,7 @@ class MelSpectrogram(torch.nn.Module):
         mel_fmin: int = 0,
         mel_fmax: int = None,
         clamp: float = 1e-5,
-        device = torch.device("cpu"),
+        device=torch.device("cpu"),
     ):
         super().__init__()
         if n_fft is None:
@@ -45,7 +45,7 @@ class MelSpectrogram(torch.nn.Module):
             hop_length=hop_length,
             win_length=win_length,
             window="hann",
-            use_torch_stft="privateuseone" not in str(device)
+            use_torch_stft="privateuseone" not in str(device),
         ).to(device)
 
     def forward(
diff --git a/rvc/f0/stft.py b/rvc/f0/stft.py
index 262f3c7..b84daa8 100644
--- a/rvc/f0/stft.py
+++ b/rvc/f0/stft.py
@@ -14,7 +14,7 @@ class STFT(torch.nn.Module):
         hop_length=512,
         win_length: Optional[int] = None,
         window="hann",
-        use_torch_stft = True,
+        use_torch_stft=True,
     ):
         """
         This module implements an STFT using 1D convolution and 1D transpose convolutions.