2 files changed, 11 insertions, 2 deletions
diff --git a/tests/storage/test_parser.cpp b/tests/storage/test_parser.cpp
index 7191080..a0045e6 100644
--- a/tests/storage/test_parser.cpp
+++ b/tests/storage/test_parser.cpp
@@ -24,7 +24,10 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include "pinyin_internal.h"
+#include <errno.h>
+#include "pinyin_base.h"
+
+using namespace pinyin;
 
 static const char *help_msg =
     "Usage:\n"
diff --git a/tests/storage/test_pinyin_table.cpp b/tests/storage/test_pinyin_table.cpp
index 659e236..42da434 100644
--- a/tests/storage/test_pinyin_table.cpp
+++ b/tests/storage/test_pinyin_table.cpp
@@ -1,6 +1,12 @@
 #include "timer.h"
 #include <string.h>
-#include "pinyin_internal.h"
+#include <errno.h>
+#include "novel_types.h"
+#include "pinyin_base.h"
+#include "pinyin_large_table.h"
+#include "phrase_index.h"
+
+using namespace pinyin;
 
 size_t bench_times = 1000;