diff --git a/paddle/fluid/framework/framework.proto b/paddle/fluid/framework/framework.proto
index 4d2bf4e4ea36e4b39cfda4d1b920ddff9f2e6397..8c57ae3e993bd7c2369996070e920acaf9c506d7 100644
--- a/paddle/fluid/framework/framework.proto
+++ b/paddle/fluid/framework/framework.proto
@@ -117,7 +117,7 @@ message VarType {
   required Type type = 1;
 
   message TensorDesc {
-    required DataType data_type = 1;
+    required Type data_type = 1;
     repeated int64 dims = 2; // [UNK, 640, 480] is saved as [-1, 640, 480]
   }
   optional TensorDesc selected_rows = 2;