summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lexer.c4
1 files changed, 2 insertions, 2 deletions
diff --git a/lexer.c b/lexer.c
index ce96a0d..f8a6ae1 100644
--- a/lexer.c
+++ b/lexer.c
@@ -181,10 +181,10 @@ get_digits:
Rune name[2] = {peek, 0};
stmt->toks[stmt->ntoks].tag = NameTag;
stmt->toks[stmt->ntoks].name = runestrdup(name);
- }else if(isalpharune(peek)){
+ }else if(isalpharune(peek) || peek == '_'){
Rune buf[64];
Rune *p = buf;
- while(isalpharune(peek) || isdigitrune(peek)){
+ while(isalpharune(peek) || isdigitrune(peek) || peek == '_'){
*p++ = peek;
peek = getrune(input);
}