diff --git a/paddle/operators/sigmoid_op.h b/paddle/operators/sigmoid_op.h
index 611c236675244383a587d8cfd41a7765272fc1da..7af879b2091e4a7f80a3a64be029394156650c23 100644
--- a/paddle/operators/sigmoid_op.h
+++ b/paddle/operators/sigmoid_op.h
@@ -42,9 +42,9 @@ class SigmoidKernel : public framework::OpKernel {
 };
 
 template <typename Place, typename T>
-class SigmoidGradKernel : public OpKernel {
+class SigmoidGradKernel : public framework::OpKernel {
  public:
-  void Compute(const ExecutionContext& context) const override {
+  void Compute(const framework::ExecutionContext& context) const override {
     auto Y_t = context.Input<Tensor>("Y");
     auto dY_t = context.Input<Tensor>(framework::GradVarName("Y"));
     auto dX_t = context.Output<Tensor>(framework::GradVarName("X"));