1 files changed, 7 insertions, 2 deletions
diff --git a/tests/storage/test_chewing_table.cpp b/tests/storage/test_chewing_table.cpp
index b600af8..d79a446 100644
--- a/tests/storage/test_chewing_table.cpp
+++ b/tests/storage/test_chewing_table.cpp
@@ -76,7 +76,12 @@ int main(int argc, char * argv[]) {
 
         guint32 start = record_time();
         PhraseIndexRanges ranges;
-        for (size_t i = 0; i < PHRASE_INDEX_LIBRARY_COUNT; ++i) {
+        memset(ranges, 0, sizeof(PhraseIndexRanges));
+
+        guint8 min_index, max_index;
+        phrase_index.get_sub_phrase_range(min_index, max_index);
+
+        for (size_t i = min_index; i < max_index; ++i) {
             ranges[i] = g_array_new(FALSE, FALSE, sizeof(PhraseIndexRange));
         }
 
@@ -84,7 +89,7 @@ int main(int argc, char * argv[]) {
             largetable.search(keys->len, (ChewingKey *)keys->data, ranges);
         }
 
-        for (size_t i = 0; i < PHRASE_INDEX_LIBRARY_COUNT; ++i) {
+        for (size_t i = min_index; i < max_index; ++i) {
             g_array_set_size(ranges[i], 0);
         }
         print_time(start, bench_times);