1 files changed, 4 insertions, 4 deletions

diff --git a/src/yuescript/yue_parser.cpp b/src/yuescript/yue_parser.cpp
index 8325366..258997e 100755
--- a/src/yuescript/yue_parser.cpp
+++ b/src/yuescript/yue_parser.cpp
@@ -237,7 +237,8 @@ YueParser::YueParser() {
         Switch = Space >> key("switch") >> disable_do(Exp) >> -(Space >> key("do"))
                 >> -Space >> Break >> SwitchBlock;
 
-        IfCond = disable_do_chain(disable_arg_table_block(Exp >> -Assign));
+        assignment = ExpList >> Assign;
+        IfCond = disable_do_chain(disable_arg_table_block(assignment | Exp));
         IfElseIf = -(Break >> *EmptyLine >> CheckIndent) >> Space >> key("elseif") >> IfCond >> plain_body_with("then");
         IfElse = -(Break >> *EmptyLine >> CheckIndent) >> Space >> key("else") >> plain_body;
         IfType = (expr("if") | expr("unless")) >> not_(AlphaNum);
@@ -619,10 +620,9 @@ YueParser::YueParser() {
 
         ExpListAssign = ExpList >> -(Update | Assign);
 
-        if_line = Space >> key("if") >> Exp >> -Assign;
-        unless_line = Space >> key("unless") >> Exp;
+        if_line = Space >> IfType >> IfCond;
 
-        statement_appendix = (if_line | unless_line | CompInner) >> Space;
+        statement_appendix = (if_line | CompInner) >> Space;
         statement_sep = and_(*SpaceBreak >> CheckIndent >> Space >> (set("($'\"") | expr("[[") | expr("[=")));
         Statement = Space >> (
                 Import | While | Repeat | For | ForEach |