diff --git a/tests/models/demucs_test.py b/tests/models/demucs_test.py
index e1203b7..0472cc2 100644
--- a/tests/models/demucs_test.py
+++ b/tests/models/demucs_test.py
@@ -15,7 +15,9 @@ def vctk_dataset():
         test_clean="clean_testset_wav",
         test_noisy="noisy_testset_wav",
     )
-    dataset = MayaDataset(name="vctk", root_dir=root_dir, files=files)
+    dataset = MayaDataset(
+        name="vctk", root_dir=root_dir, files=files, sampling_rate=16000
+    )
     return dataset
 
 
diff --git a/tests/models/test_dccrn.py b/tests/models/test_dccrn.py
index bc2a039..b309cb7 100644
--- a/tests/models/test_dccrn.py
+++ b/tests/models/test_dccrn.py
@@ -15,7 +15,9 @@ def vctk_dataset():
         test_clean="clean_testset_wav",
         test_noisy="noisy_testset_wav",
     )
-    dataset = MayaDataset(name="vctk", root_dir=root_dir, files=files)
+    dataset = MayaDataset(
+        name="vctk", root_dir=root_dir, files=files, sampling_rate=16000
+    )
     return dataset
 
 
diff --git a/tests/models/test_waveunet.py b/tests/models/test_waveunet.py
index bc250d1..ca0af0a 100644
--- a/tests/models/test_waveunet.py
+++ b/tests/models/test_waveunet.py
@@ -15,7 +15,9 @@ def vctk_dataset():
         test_clean="clean_testset_wav",
         test_noisy="noisy_testset_wav",
     )
-    dataset = MayaDataset(name="vctk", root_dir=root_dir, files=files)
+    dataset = MayaDataset(
+        name="vctk", root_dir=root_dir, files=files, sampling_rate=16000
+    )
     return dataset