diff --git a/funcodec/models/codec_basic.py b/funcodec/models/codec_basic.py
index 6932d58..51c5f74 100644
--- a/funcodec/models/codec_basic.py
+++ b/funcodec/models/codec_basic.py
@@ -32,7 +32,7 @@ def __init__(
         n_mel_channels=80,
         mel_fmin=0.0,
         mel_fmax=None,
-        device='cuda'
+        device='cpu'
     ):
         super().__init__()
         ##############################################
@@ -42,7 +42,7 @@ def __init__(
         mel_basis = librosa_mel_fn(
             sr=sampling_rate, n_fft=n_fft, n_mels=n_mel_channels, fmin=mel_fmin, fmax=mel_fmax
         )
-        mel_basis = torch.from_numpy(mel_basis).cuda().float()
+        mel_basis = torch.from_numpy(mel_basis).to(device).float()
         self.register_buffer("mel_basis", mel_basis)
         self.register_buffer("window", window)
         self.n_fft = n_fft
diff --git a/setup.py b/setup.py
index c535955..ab9f8bb 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@
         # ENH
         # "ci_sdr",
         "pytorch_wpe",
-        "editdistance==0.5.2",
+        "editdistance",
         "tensorboard>=1.15",
         "g2p",
         # PAI
@@ -156,4 +156,4 @@
         "License :: OSI Approved :: Apache Software License",
         "Topic :: Software Development :: Libraries :: Python Modules",
     ],
-)
\ No newline at end of file
+)