From 76cb3b7874c86eec9c610d74d296d0adfeb4315e Mon Sep 17 00:00:00 2001
From: gingerBill <bill@gingerbill.org>
Date: Mon, 27 Feb 2023 16:15:19 +0000
Subject: [PATCH] Add better fallback for `ast_token`

---
 src/parser_pos.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/parser_pos.cpp b/src/parser_pos.cpp
index 8a0c3eca6..1274f05a0 100644
--- a/src/parser_pos.cpp
+++ b/src/parser_pos.cpp
@@ -23,6 +23,9 @@ gb_internal Token ast_token(Ast *node) {
 		if (node->SelectorExpr.expr != nullptr) {
 			return ast_token(node->SelectorExpr.expr);
 		}
+		if (node->SelectorExpr.selector != nullptr) {
+			return ast_token(node->SelectorExpr.selector);
+		}
 		return node->SelectorExpr.token;
 	case Ast_SelectorCallExpr:
 		if (node->SelectorCallExpr.expr != nullptr) {