diff --git a/auto_transformers.py b/auto_transformers.py
index 8f51d30..3d14f52 100644
--- a/auto_transformers.py
+++ b/auto_transformers.py
@@ -45,7 +45,7 @@ class AutoTransformers(NNOperator):
             self.model = AutoModel.from_pretrained(model_name)
             self.model.eval()
         except Exception as e:
-            model_list = get_model_list()
+            model_list = self.get_model_name()
             if model_name not in model_list:
                 log.error(f"Invalid model name: {model_name}. Supported model names: {model_list}")
             else: