diff --git a/tools/program.py b/tools/program.py
index d68e123a85cf7d941db9f66b8fc5b9bac0064d50..0cc60c3877fa4fd3b9eb2ec12f3a0c5d378db15b 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()]))