diff --git a/templates/repo/modelsafety/new.tmpl b/templates/repo/modelsafety/new.tmpl
index db1cce846..22b623d24 100644
--- a/templates/repo/modelsafety/new.tmpl
+++ b/templates/repo/modelsafety/new.tmpl
@@ -352,7 +352,7 @@
     })
 
     ;(function() {
-        var SPECS = {{ .benchmark_specs }} || {{ .Specs }};
+        var SPECS = {{ .train_specs }} || {{ .Specs }};
         var showPoint = {{ .CloudBrainPaySwitch }};
         window.renderSpecsSelect($('#__specs__'), SPECS, showPoint, {
             gpu_memory: {{$.i18n.Tr "cloudbrain.gpu_memory"}},