diff --git a/test/test_cuda/test_main_func.py b/test/test_cuda/test_main_func.py
index 523fe6784..571fc10f5 100644
--- a/test/test_cuda/test_main_func.py
+++ b/test/test_cuda/test_main_func.py
@@ -212,7 +212,6 @@ def test_low_cpu_mem_usage(self):
             group_size=group_size,
             iters=2,
             seqlen=10,
-            dataset=self.llm_dataloader,
             low_cpu_mem_usage=True,
         )
         autoround.quantize_and_save(output_dir=quantized_model_path, format="auto_round")