diff --git a/tests/test_auto3dseg.py b/tests/test_auto3dseg.py
index deb54b7a4b..53f25051ec 100644
--- a/tests/test_auto3dseg.py
+++ b/tests/test_auto3dseg.py
@@ -184,7 +184,7 @@ def test_data_analyzer_cpu(self, input_params):
         )
 
         analyser = DataAnalyzer(
-            self.datalist_file, self.dataroot_dir, output_path=self.datastat_file, label_key=label_key
+            self.datalist_file, self.dataroot_dir, output_path=self.datastat_file, label_key=label_key, device=device
         )
         datastat = analyser.get_all_case_stats()