1 files changed, 3 insertions, 1 deletions
diff --git a/src/lexer/lexer.c b/src/lexer/lexer.c
index 183bd76..9be4cb1 100644
--- a/src/lexer/lexer.c
+++ b/src/lexer/lexer.c
@@ -63,7 +63,8 @@ lex_identifier(struct lexer *ctx, struct token *tok)
     /* Look for a keyword matching the identifier */
     kwd = keywords_find(tok);
     if (kwd != NULL) {
-        tok->kind = kwd->value;
+        tok->kind = kwd->tok_kind;
+        tok->flags |= kwd->tok_flags;
     } else {
         tok->kind = TK_IDENTIFIER;
     }
@@ -78,6 +79,7 @@ lexer_next(struct lexer *ctx, struct token *tok)
 
     skip_whitespace(ctx);
 
+    tok->flags = TF_NONE;
     tok->pos = ctx->pos;
     tok->line = ctx->line;
     tok->col = (int)(tok->pos - ctx->line_start) + 1;