1 files changed, 4 insertions, 1 deletions
diff --git a/utils/training/export_k_mixture_model.cpp b/utils/training/export_k_mixture_model.cpp
index 95ea1ef..4b3d993 100644
--- a/utils/training/export_k_mixture_model.cpp
+++ b/utils/training/export_k_mixture_model.cpp
@@ -130,7 +130,10 @@ int main(int argc, char * argv[]){
         exit(ENOENT);
 
     KMixtureModelBigram bigram(K_MIXTURE_MODEL_MAGIC_NUMBER);
-    bigram.attach(k_mixture_model_filename, ATTACH_READONLY);
+    if (!bigram.attach(k_mixture_model_filename, ATTACH_READONLY)) {
+        fprintf(stderr, "open %s failed.\n", k_mixture_model_filename);
+        exit(ENOENT);
+    }
 
     print_k_mixture_model_magic_header(output, &bigram);
     print_k_mixture_model_array_headers(output, &bigram, &phrase_index);