diff --git a/paddle/fluid/lite/api/cxx_api_bin.cc b/paddle/fluid/lite/api/cxx_api_bin.cc
index f17f7a7d5faf31e453a5759320884399abdbc47f..b315030ed7ed7560251c163a9106717fe6d11548 100644
--- a/paddle/fluid/lite/api/cxx_api_bin.cc
+++ b/paddle/fluid/lite/api/cxx_api_bin.cc
@@ -69,7 +69,7 @@ USE_LITE_OP(fetch);
 USE_LITE_OP(io_copy);
 
 USE_LITE_OP(conv2d);
-// USE_LITE_OP(batch_norm);
+USE_LITE_OP(batch_norm);
 USE_LITE_OP(relu);
 USE_LITE_OP(depthwise_conv2d);
 USE_LITE_OP(pool2d);
diff --git a/paddle/fluid/lite/operators/batch_norm_op.cc b/paddle/fluid/lite/operators/batch_norm_op.cc
index e974d0134dad93a2241c265687a190b10d5ff85d..b6ef87732defcb879f655c3177c00725bf57cfed 100644
--- a/paddle/fluid/lite/operators/batch_norm_op.cc
+++ b/paddle/fluid/lite/operators/batch_norm_op.cc
@@ -82,7 +82,7 @@ bool BatchNormOp::AttachImpl(const cpp::OpDesc &op_desc, lite::Scope *scope) {
   param_.variance =
       scope->FindVar(op_desc.Input("Variance").front())->GetMutable<Tensor>();
   param_.y = scope->FindVar(op_desc.Output("Y").front())->GetMutable<Tensor>();
-  param_.is_test = op_desc.GetAttr<bool>("is_test");
+  param_.is_test = op_desc.GetAttr<int>("is_test");
   param_.use_global_stats = op_desc.GetAttr<bool>("use_global_stats");
   if (!param_.is_test) {
     param_.mean_out =