diff --git a/paddle/fluid/operators/assign_value_op.h b/paddle/fluid/operators/assign_value_op.h
index 6a62ca53d4782ce68dae249aeeb3e514d32e6d6d..cdcd7c406adcb77a76f7322bfb071ba20c637a45 100644
--- a/paddle/fluid/operators/assign_value_op.h
+++ b/paddle/fluid/operators/assign_value_op.h
@@ -30,10 +30,10 @@ class AssignValueKernel : public framework::OpKernel<T> {
     int dtype = ctx.Attr<int>("dtype");
     const char* value_name = nullptr;
     switch (dtype) {
-      case framework::proto::VarType::Type::INT32:
+      case framework::proto::VarType::INT32:
         value_name = "int32_values";
         break;
-      case framework::proto::VarType::Type::FP32:
+      case framework::proto::VarType::FP32:
         value_name = "fp32_values";
         break;
       default: