diff --git a/models/vista3d/scripts/evaluator.py b/models/vista3d/scripts/evaluator.py
index a95bdf30..3f0b7777 100644
--- a/models/vista3d/scripts/evaluator.py
+++ b/models/vista3d/scripts/evaluator.py
@@ -167,7 +167,7 @@ def check_prompts_format(self, label_prompt, points, point_labels):
                 if not np.all([(x < 255).item() for x in label_prompt]):
                     raise ValueError("Current bundle only supports label prompt smaller than 255.")
                 if points is None:
-                    supported_list = list(set([i + 1 for i in range(132)]) - set([16, 18, 129, 130, 131]))
+                    supported_list = set(i + 1 for i in range(132)) - {16, 18, 129, 130, 131}
                     if not np.all([x in supported_list for x in label_prompt]):
                         raise ValueError("Undefined label prompt detected. Provide point prompts for zero-shot.")
             else: