diff --git a/new_datasets/iterate_hf_dataset.py b/new_datasets/iterate_hf_dataset.py
index fea5aefdbb72ddad2486e3ac29497a6b032879c2..b4285fa1498c4b843a0dc530257c3e022d38d194 100644
--- a/new_datasets/iterate_hf_dataset.py
+++ b/new_datasets/iterate_hf_dataset.py
@@ -23,4 +23,4 @@ if __name__ == '__main__':
     parser.add_argument("--split_name")
     parser.add_argument("--items")
     args = parser.parse_args()
-    iterate_dataset(args.dataset_path, args.dataset_name, args.cache_dir, int(args.items))
+    iterate_dataset(args.dataset_path, args.dataset_name, args.cache_dir, args.split_name, int(args.items))