update training tools

1 files changed, 2 insertions, 2 deletions

diff --git a/utils/training/gen_k_mixture_model.cpp b/utils/training/gen_k_mixture_model.cpp
index 2dfb3d1..0679a7f 100644
--- a/utils/training/gen_k_mixture_model.cpp
+++ b/utils/training/gen_k_mixture_model.cpp
@@ -329,7 +329,7 @@ int main(int argc, char * argv[]){
         exit(EINVAL);
     }
 
-    SystemTableInfo system_table_info;
+    SystemTableInfo2 system_table_info;
 
     bool retval = system_table_info.load(SYSTEM_TABLE_INFO);
     if (!retval) {
@@ -345,7 +345,7 @@ int main(int argc, char * argv[]){
     FacadePhraseIndex phrase_index;
 
     const pinyin_table_info_t * phrase_files =
-        system_table_info.get_table_info();
+        system_table_info.get_default_tables();
 
     if (!load_phrase_index(phrase_files, &phrase_index))
         exit(ENOENT);