diff --git a/core/paddlefl_mpc/operators/mpc_elementwise_add_op.h b/core/paddlefl_mpc/operators/mpc_elementwise_add_op.h
index d30f3aa998a53922bfb2557b5ff1b2c1c200f7c5..e25e8709f712487186707033490cf89d998c1cf4 100644
--- a/core/paddlefl_mpc/operators/mpc_elementwise_add_op.h
+++ b/core/paddlefl_mpc/operators/mpc_elementwise_add_op.h
@@ -106,9 +106,9 @@ template <typename DeviceContext, typename T>
 class MpcElementwiseAddKernel : public MpcOpKernel<T> {
 public:
   void ComputeImpl(const framework::ExecutionContext &ctx) const override {
-    auto *in_x_t = ctx.Input<LoDTensor>("X");
-    auto *in_y_t = ctx.Input<LoDTensor>("Y");
-    auto *out_t = ctx.Output<LoDTensor>("Out");
+    auto *in_x_t = ctx.Input<Tensor>("X");
+    auto *in_y_t = ctx.Input<Tensor>("Y");
+    auto *out_t = ctx.Output<Tensor>("Out");
 
     int axis = ctx.Attr<int>("axis");
 
diff --git a/python/setup.py b/python/setup.py
index d9359e58b78b9f09405cff534b347a04b928a513..79f73d506d71d9771ec06625c2d0489c5190ba13 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -22,7 +22,7 @@ import platform
 
 from setuptools import find_packages
 from setuptools import setup
-
+from paddle_fl.version import fl_version
 
 def python_version():
     """