X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/5602b1eebd5c9733b58795485c6a592c6e77b4d8..9417a6380a3b828f44584009384d8686cd8d7e59:/Cycript.yy.in

diff --git a/Cycript.yy.in b/Cycript.yy.in
index 1345060..465b32b 100644
--- a/Cycript.yy.in
+++ b/Cycript.yy.in
@@ -536,11 +536,11 @@ LexNoBrace
     ;
 
 LexNoFunction
-    : { driver.no_.Function = true; }
+    : { if (yychar == yyempty_) driver.no_.Function = true; else if (yychar == token::Function) yychar = token::Function_; }
     ;
 
 LexNoAtImplementation
-    : { driver.no_.AtImplementation = true; }
+    : { if (yychar == yyempty_) driver.no_.AtImplementation = true; else if (yychar == token::AtImplementation) yychar = token::AtImplementation_; }
     ;
 
 LexSetStatement