1 files changed, 3 insertions, 2 deletions
diff --git a/utils/dict.h b/utils/dict.h
index 75ea3def..33cca6cf 100644
--- a/utils/dict.h
+++ b/utils/dict.h
@@ -1,7 +1,7 @@
 #ifndef DICT_H_
 #define DICT_H_
 
-
+#include <iostream>
 #include <cassert>
 #include <cstring>
 
@@ -73,7 +73,8 @@ class Dict {
 
   inline const std::string& Convert(const WordID& id) const {
     if (id == 0) return b0_;
-    assert(id <= (int)words_.size());
+    //assert(id <= (int)words_.size());
+    if (id < 0 || id > (int)words_.size()) return b0_;
     return words_[id-1];
   }