diff --git a/booster/xgboost.cpp b/booster/xgboost.cpp
index 6cac83488..381067d08 100644
--- a/booster/xgboost.cpp
+++ b/booster/xgboost.cpp
@@ -9,7 +9,7 @@
 #include <climits>
 #include "xgboost.h"
 #include "../utils/xgboost_utils.h"
-#include "../gbm-base/xgboost_base_model.h"
+#include "xgboost_gbmbase.h"
 namespace xgboost{
     namespace booster{
         /*!