diff --git a/runtime/core/frontend/wav.h b/runtime/core/frontend/wav.h
index 2e83b9e..791a128 100644
--- a/runtime/core/frontend/wav.h
+++ b/runtime/core/frontend/wav.h
@@ -137,8 +137,8 @@ class WavReader {
 
 class WavWriter {
  public:
-  WavWriter(const float* data, int num_samples, int num_channel, int sample_rate,
-            int bits_per_sample)
+  WavWriter(const float* data, int num_samples, int num_channel,
+            int sample_rate, int bits_per_sample)
       : data_(data),
         num_samples_(num_samples),
         num_channel_(num_channel),
diff --git a/runtime/core/kws/keyword_spotting.h b/runtime/core/kws/keyword_spotting.h
index 8845913..14bf732 100644
--- a/runtime/core/kws/keyword_spotting.h
+++ b/runtime/core/kws/keyword_spotting.h
@@ -16,6 +16,7 @@
 #ifndef KWS_KEYWORD_SPOTTING_H_
 #define KWS_KEYWORD_SPOTTING_H_
 
+#include <memory>
 #include <string>
 #include <vector>