3 files changed, 6 insertions, 4 deletions

diff --git a/src/yuescript/yue_ast.h b/src/yuescript/yue_ast.h
index 6b2b7f8..6d90add 100644
--- a/src/yuescript/yue_ast.h
+++ b/src/yuescript/yue_ast.h
@@ -80,6 +80,7 @@ class Macro_t;
 class In_t;
 class NormalDef_t;
 class SpreadListExp_t;
+class Comprehension_t;
 } // namespace yue
 
 AST_LEAF(Num)
@@ -171,7 +172,7 @@ AST_END(CloseAttrib, "close"sv)
 AST_NODE(LocalAttrib)
         ast_sel<true, ConstAttrib_t, CloseAttrib_t> attrib;
         ast_ptr<true, Seperator_t> sep;
-        ast_sel_list<true, Variable_t, SimpleTable_t, TableLit_t> leftList;
+        ast_sel_list<true, Variable_t, SimpleTable_t, TableLit_t, Comprehension_t> leftList;
         ast_ptr<true, Assign_t> assign;
         AST_MEMBER(LocalAttrib, &attrib, &sep, &leftList, &assign)
 AST_END(LocalAttrib, "local_attrib"sv)
diff --git a/src/yuescript/yue_compiler.cpp b/src/yuescript/yue_compiler.cpp
index d43f79b..f163bdf 100644
--- a/src/yuescript/yue_compiler.cpp
+++ b/src/yuescript/yue_compiler.cpp
@@ -75,7 +75,7 @@ static std::unordered_set<std::string> Metamethods = {
         "close"s // Lua 5.4
 };
 
-const std::string_view version = "0.21.5"sv;
+const std::string_view version = "0.21.6"sv;
 const std::string_view extension = "yue"sv;
 
 class CompileError : public std::logic_error {
@@ -9376,7 +9376,8 @@ private:
                                         case id<SimpleTable_t>():
                                                 value->item.set(item);
                                                 break;
-                                        case id<TableLit_t>(): {
+                                        case id<TableLit_t>():
+                                        case id<Comprehension_t>(): {
                                                 auto simpleValue = item->new_ptr<SimpleValue_t>();
                                                 simpleValue->value.set(item);
                                                 value->item.set(simpleValue);
diff --git a/src/yuescript/yue_parser.cpp b/src/yuescript/yue_parser.cpp
index 3557534..4e6105b 100644
--- a/src/yuescript/yue_parser.cpp
+++ b/src/yuescript/yue_parser.cpp
@@ -286,7 +286,7 @@ YueParser::YueParser() {
 
         ConstAttrib = key("const");
         CloseAttrib = key("close");
-        local_const_item = Variable | SimpleTable | TableLit;
+        local_const_item = Variable | SimpleTable | TableLit | Comprehension;
         LocalAttrib = (
                 ConstAttrib >> Seperator >> space >> local_const_item >> *(space >> ',' >> space >> local_const_item) |
                 CloseAttrib >> Seperator >> space >> Variable >> *(space >> ',' >> space >> Variable)