1 files changed, 3 insertions, 3 deletions
diff --git a/src/query.h b/src/query.h
index b5b3b0fc..aca31b61 100644
--- a/src/query.h
+++ b/src/query.h
@@ -222,14 +222,14 @@ public:
       TRACE_DTOR(query_t::lexer_t);
     }
 
-    token_t next_token();
+    token_t next_token(token_t::kind_t tok_context = token_t::UNKNOWN);
     void    push_token(token_t tok) {
       assert(token_cache.kind == token_t::UNKNOWN);
       token_cache = tok;
     }
-    token_t peek_token() {
+    token_t peek_token(token_t::kind_t tok_context = token_t::UNKNOWN) {
       if (token_cache.kind == token_t::UNKNOWN)
-        token_cache = next_token();
+        token_cache = next_token(tok_context);
       return token_cache;
     }
   };