diff --git a/tools/program.py b/tools/program.py
index d68e123a85..0cc60c3877 100755
--- a/tools/program.py
+++ b/tools/program.py
@@ -276,7 +276,7 @@ def train(config,
                     valid_dataloader,
                     post_process_class,
                     eval_class,
-                    "table",
+                    model_type,
                     use_srn=use_srn)
                 cur_metric_str = 'cur metric, {}'.format(', '.join(
                     ['{}: {}'.format(k, v) for k, v in cur_metric.items()]))