diff --git a/src/common/ast.cpp b/src/common/ast.cpp
index ac7d00138013423ca9b303e07be8c88e1af642e4..42439683015781ace95cf7905dc11175a6c906dc 100644
--- a/src/common/ast.cpp
+++ b/src/common/ast.cpp
@@ -119,7 +119,7 @@ AST::transform_node_iter(syntax_tree_node *n) {
         else
             node->type = TYPE_FLOAT;
         node->id = n->children[1]->name;
-        if (n->children_num == 3)
+        if (n->children_num > 2)
             node->isarray = true;
         return node;
     } else if (_STR_EQ(n->name, "compound-stmt")) {