diff --git a/core/nbv_dataset.py b/core/nbv_dataset.py
index ca9c0c7..36f4ed1 100644
--- a/core/nbv_dataset.py
+++ b/core/nbv_dataset.py
@@ -48,7 +48,7 @@ class NBVReconstructionDataset(BaseDataset):
             for line in f:
                 scene_name = line.strip()
                 scene_name_list.append(scene_name)
-        return scene_name_list
+        return scene_name_list[:10]
 
     def get_datalist(self):
         datalist = []