diff --git a/pytorch_translate/cpp/DbPredictor.cpp b/pytorch_translate/cpp/DbPredictor.cpp
index 1745848e..281cb540 100644
--- a/pytorch_translate/cpp/DbPredictor.cpp
+++ b/pytorch_translate/cpp/DbPredictor.cpp
@@ -1,6 +1,6 @@
 #include "DbPredictor.h"
 
-#include <caffe2/core/predictor_utils.h>
+#include <caffe2/predictor/predictor_utils.h>
 
 namespace pytorch {
 namespace translate {
diff --git a/pytorch_translate/cpp/DbPredictor.h b/pytorch_translate/cpp/DbPredictor.h
index 8650bd2e..eeee7c63 100644
--- a/pytorch_translate/cpp/DbPredictor.h
+++ b/pytorch_translate/cpp/DbPredictor.h
@@ -4,8 +4,8 @@
 
 #include <string>
 
-#include <caffe2/core/predictor.h>
 #include <caffe2/core/workspace.h>
+#include <caffe2/predictor/predictor.h>
 
 namespace pytorch {
 namespace translate {