diff --git a/tests/trainers/test_image_denoise_trainer.py b/tests/trainers/test_image_denoise_trainer.py
index 0bcb8930..68ddf616 100644
--- a/tests/trainers/test_image_denoise_trainer.py
+++ b/tests/trainers/test_image_denoise_trainer.py
@@ -33,11 +33,13 @@ class ImageDenoiseTrainerTest(unittest.TestCase):
         dataset_train = MsDataset.load(
             'SIDD',
             namespace='huizheng',
+            subset_name='default',
             split='validation',
             download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds
         dataset_val = MsDataset.load(
             'SIDD',
             namespace='huizheng',
+            subset_name='default',
             split='test',
             download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds
         self.dataset_train = SiddImageDenoisingDataset(