diff --git a/new_datasets/download_hf_dataset.py b/new_datasets/download_hf_dataset.py
index 1f6effc4ce190ee090f0b3877f509f02863a20b2..45e472c69e5ec5353760099a2cba668c14ea1ce2 100644
--- a/new_datasets/download_hf_dataset.py
+++ b/new_datasets/download_hf_dataset.py
@@ -5,7 +5,7 @@ import datasets
 
 
 def download_dataset(dataset_path: str, dataset_name: Optional[str], cache_dir: str):
-    dataset = datasets.load_dataset(dataset_path, dataset_name, cache_dir=cache_dir)
+    dataset = datasets.load_dataset(dataset_path, dataset_name if len(dataset_name) > 0 else None, cache_dir=cache_dir)
     print(dataset)
 
 
@@ -15,4 +15,4 @@ if __name__ == '__main__':
     parser.add_argument("--dataset_name")
     parser.add_argument("--cache_dir")
     args = parser.parse_args()
-    download_dataset(args.dataset, args.asr)
+    download_dataset(args.dataset_path, args.dataset_name, args.cache_dir)