diff --git a/pdseg/vis.py b/pdseg/vis.py
index 6d6a457646e545e3b879b5bc55ea4e2b2848949a..7050b0365bc8374052cfe56840dc77fff967913e 100644
--- a/pdseg/vis.py
+++ b/pdseg/vis.py
@@ -93,7 +93,7 @@ def visualize(cfg,
               local_test=False,
               **kwargs):
     if vis_file_list is None:
-        vis_file_list = cfg.DATASET.TEST_FILE_LIST
+        vis_file_list = cfg.DATASET.VIS_FILE_LIST
     dataset = SegDataset(
         file_list=vis_file_list,
         mode=ModelPhase.VISUAL,