diff --git a/benchmark_model.cc b/benchmark_model.cc
index f41dcd02c476b156d876c6d989b383eb1721529c..316de8273f377ab52282b7b17d32e7a4fd0778d9 100644
--- a/benchmark_model.cc
+++ b/benchmark_model.cc
@@ -3,7 +3,7 @@
 //
 
 #include "gflags/gflags.h"
-#include "mace/core/public/mace.h"
+#include "mace/public/mace.h"
 #include "benchmark/stat_summarizer.h"
 
 #include <cstdlib>
diff --git a/stat_summarizer.cc b/stat_summarizer.cc
index 065c627c0e378b3b73363aef83ec6cb46298e2ad..b95bac953ef9e02cc63cdcf710d8ab6fb3c6a6a4 100644
--- a/stat_summarizer.cc
+++ b/stat_summarizer.cc
@@ -3,7 +3,7 @@
 //
 
 #include "benchmark/stat_summarizer.h"
-#include "mace/core/public/mace.h"
+#include "mace/public/mace.h"
 
 #include <iomanip>
 #include <queue>