diff --git a/mace/ops/reshape_test.cc b/mace/ops/reshape_test.cc
index 4e48d384162e3401e083384a0d1176a7227114a3..91c0f82b7ae24c7da41ca5e504fa3aace600f29f 100644
--- a/mace/ops/reshape_test.cc
+++ b/mace/ops/reshape_test.cc
@@ -36,9 +36,10 @@ void TestReshape(const std::vector<index_t> &org_shape,
 
   // Add input data
   net.AddRandomInput<DeviceType::CPU, float>("Input", org_shape);
-  net.AddInputFromArray<DeviceType::CPU, int32_t>("Shape",
-                                                  {output_shape.size()},
-                                                  output_shape);
+  net.AddInputFromArray<DeviceType::CPU, int32_t>(
+      "Shape",
+      {static_cast<index_t>(output_shape.size())},
+      output_shape);
 
   // Run
   net.RunOp();