From 2bd236dd1f3de1d6266406f93528040299baaf22 Mon Sep 17 00:00:00 2001
From: Valk Richard Li <48872266+ValKmjolnir@users.noreply.github.com>
Date: Mon, 14 Oct 2019 22:53:30 +0800
Subject: [PATCH] Bug fixed

---
 version1.0/ast_generator.h | 2 ++
 version1.0/nasal_parser.h  | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/version1.0/ast_generator.h b/version1.0/ast_generator.h
index 72127a7..41a523f 100644
--- a/version1.0/ast_generator.h
+++ b/version1.0/ast_generator.h
@@ -138,6 +138,7 @@ abstract_syntax_tree generator::statements_block()
 				}
 				else
 				{
+					parse.push(this_token);
 					node.add_statement(in_curve_calc_expr());
 				}
 				check_semi_at_end();
@@ -1087,6 +1088,7 @@ void generator::gen_main_work()
 				}
 				else
 				{
+					parse.push(this_token);
 					root.add_statement(in_curve_calc_expr());
 				}
 				check_semi_at_end();
diff --git a/version1.0/nasal_parser.h b/version1.0/nasal_parser.h
index 7b143c7..d1518b8 100644
--- a/version1.0/nasal_parser.h
+++ b/version1.0/nasal_parser.h
@@ -179,6 +179,7 @@ void nasal_parser::statements_block()
 				}
 				else
 				{
+					parse.push(this_token);
 					in_curve_calc_expr();
 				}
 				check_semi_at_end();
@@ -1360,6 +1361,7 @@ void nasal_parser::parse_main_work()
 				}
 				else
 				{
+					parse.push(this_token);
 					in_curve_calc_expr();
 				}
 				check_semi_at_end();