diff --git a/paddle/fluid/recordio/writer.cc b/paddle/fluid/recordio/writer.cc
index 7a4143c662350158051b767db1f7eb820c7436c4..196d66edff8cc6000afcd74fb945c05dcab7106a 100644
--- a/paddle/fluid/recordio/writer.cc
+++ b/paddle/fluid/recordio/writer.cc
@@ -12,7 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 #include "paddle/fluid/recordio/writer.h"
-
+#include "paddle/fluid/platform/enforce.h"
 namespace paddle {
 namespace recordio {
 void Writer::Write(const std::string& record) {
diff --git a/paddle/fluid/recordio/writer.h b/paddle/fluid/recordio/writer.h
index 2db6f60f41acbe6b52dc47b7670fea39065be4d1..0c478d507547b10b8ebaaf5e512557a5c8c13e65 100644
--- a/paddle/fluid/recordio/writer.h
+++ b/paddle/fluid/recordio/writer.h
@@ -13,7 +13,6 @@
 // limitations under the License.
 
 #pragma once
-#include "paddle/fluid/framework/lod_tensor.h"
 #include "paddle/fluid/recordio/chunk.h"
 namespace paddle {
 namespace recordio {