diff --git a/include/tvm/relay/attrs/memory.h b/include/tvm/relay/attrs/memory.h
index b737103e715f2..85462c087cee0 100644
--- a/include/tvm/relay/attrs/memory.h
+++ b/include/tvm/relay/attrs/memory.h
@@ -35,7 +35,7 @@ namespace relay {
 
 std::vector<TensorType> FlattenTupleType(const Type& type);
 std::vector<Expr> FromTupleType(const Type& type, const Expr& expr);
-Expr ToTupleType(const Type& t, const Array<Expr>& exprs);
+Expr ToTupleType(const Type& t, const std::vector<Expr>& exprs);
 
 /*!
  * \brief Options for allocating storage.